LeetCode 算法：交换和

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第 23 天，点击查看活动详情

交换和

原题地址

给定两个整数数组，请交换一对数值（每个数组中取一个数值），使得两个数组所有元素的和相等。

返回一个数组，第一个元素是第一个数组中要交换的元素，第二个元素是第二个数组中要交换的元素。若有多个答案，返回任意一个均可。若无满足条件的数值，返回空数组。

示例:

输入: array1 = [4, 1, 2, 1, 1, 2], array2 = [3, 6, 3, 3]
输出: [1, 3]

示例:

输入: array1 = [1, 2, 3], array2 = [4, 5, 6]
输出: []

提示：

• 1 <= array1.length, array2.length <= 100000

思路分析

方法一

1. 首先计算两个数组当前的和，然后将两个数组使用 Set 进行去重；
2. 循环比较，先循环 array1，然后将 sum1 减去 array1[i]，同时给 sum2 加上 array1[i]，然后在循环 array2，此时若满足题目中的条件，则必须满足 sum1 + array2[j] === sum2 - array2[j]；
3. 按照步骤2进行判断，若满足条件，返回 [array1[i], array2[j]]；若不满足，则返回 []。

方法二

1. 第一步还是同方法一，计算和并去重；
2. 根据题目分析可以得知，sum1 - sum2 = 2 * (array1[i] - array2[j])；
3. 因此可以进行循环来寻找 array[i] - array2[j] 等于 (sum1 - sum2) / 2 的值，当然都是按照绝对值计算；
4. 找到对应的值返回，若循环完毕后没有返回值，则返回 []。

AC 代码

方法一

/**
 * @param {number[]} array1
 * @param {number[]} array2
 * @return {number[]}
 */
var findSwapValues = function(array1, array2) {
    let sum1 = array1.reduce((prev, curr) => prev + curr, 0)
    let sum2 = array2.reduce((prev, curr) => prev + curr, 0)
    
    array1 = [...new Set(array1)]
    array2 = [...new Set(array2)]
   
    for (let i = 0; i < array1.length; i++) {
        sum1 -= array1[i]
        sum2 += array1[i]
        for (let j = 0; j < array2.length; j++) {
            if (sum1 + array2[j] === sum2 - array2[j]) {
                return [array1[i], array2[j]]
            }
        }
        sum1 += array1[i]
        sum2 -= array1[i]
    }
    return []
};

结果：

• 执行结果： 通过
• 执行用时：76 ms, 在所有 JavaScript 提交中击败了71.26%的用户
• 内存消耗：45.2 MB, 在所有 JavaScript 提交中击败了66.67%的用户
• 通过测试用例：34 / 34

方法二

/**
 * @param {number[]} array1
 * @param {number[]} array2
 * @return {number[]}
 */
var findSwapValues = function(array1, array2) {
    let sum1 = array1.reduce((prev, curr) => prev + curr, 0)
    let sum2 = array2.reduce((prev, curr) => prev + curr, 0)
    
    array1 = [...new Set(array1)]
    array2 = [...new Set(array2)]
    
    let count = Math.abs(sum1 - sum2) / 2
    
    for (let i = 0; i < array1.length; i++) {
        const index = array2.findIndex(item => Math.abs(array1[i] - item) === count)
        if (index > -1) {
            return [array1[i], array2[index]]
        }
    }
    return []
};

结果：

• 执行结果： 通过
• 执行用时：72 ms, 在所有 JavaScript 提交中击败了85.06%的用户
• 内存消耗：45.2 MB, 在所有 JavaScript 提交中击败了63.22%的用户
• 通过测试用例：34 / 34

END