给定两个整数数组,请交换一对数值(每个数组中取一个数值),使得两个数组所有元素的和相等。
返回一个数组,第一个元素是第一个数组中要交换的元素,第二个元素是第二个数组中要交换的元素。若有多个答案,返回任意一个均可。若无满足条件的数值,返回空数组。
示例:
输入: array1 = [4, 1, 2, 1, 1, 2], array2 = [3, 6, 3, 3]
输出: [1, 3]
示例:
输入: array1 = [1, 2, 3], array2 = [4, 5, 6]
输出: []
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-swap-lcci
刷题就是搞数学 ,理解了题目意思就好做了、
class Solution {
public int[] findSwapValues(int[] array1, int[] array2) {
/**
*
* @思路: 假设array1的和为sum1,要交换的元素为x1. array2的和为sum2,要交换的元素为x2.
* 为了满足题目要求就有: sum1 - x1 + x2 = sum2 - x2 + x1
* => sum1 - sum2 = 2(x1 - x2)
* => (sum1 - sum2) / 2 = x1 - x2
* 现在,遍历array1,计算出x2,看看满足上述条件的x2是否存在array2中,若存在,说明找到了要交换的数组,否则返回[].
* @param {number[]} array1
* @param {number[]} array2
* @returns {number[]}
*/
int diff = Arrays.stream(array1).sum() - Arrays.stream(array2).sum();
if((diff % 2) != 0) return new int[0];
int finaldiff = diff >> 1;
Set<Integer> set = new HashSet<>();
for(int i : array2){
set.add(i);
}
for(int j: array1){
if(set.contains(j-finaldiff)){
return new int[]{j,j-finaldiff};
}
}
return new int[]{};
}
}