程序员面试金典 - 面试题 16.21. 交换和（哈希set）

1. 题目

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

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

示例:
输入: 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-swap-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

• $S1-a+b = S2-b+a \Rightarrow a -(S1-S2)/2=b$

class Solution {
public:
    vector<int> findSwapValues(vector<int>& array1, vector<int>& array2) {
    	int s1 = 0, s2 = 0, i;
    	unordered_set<int> set;
    	for(i = 0; i < array1.size(); ++i)
    		s1 += array1[i];
    	for(i = 0; i < array2.size(); ++i)
    	{
    		s2 += array2[i];
    		set.insert(array2[i]);
    	}
    	if(abs(s1-s2)%2)
    		return {};
    	int t = (s1-s2)>>1;
    	for(i = 0; i < array1.size(); ++i)
    	{
    		if(set.count(array1[i]-t))
    			return {array1[i], array1[i]-t};
    	}
    	return {};
    }
};

196 ms 33 MB