leetcode---数组系列--面试题 16.21. 交换和(哈希查找) C++

面试题 16.21. 交换和 （C++）

题目描述：

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

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

示例1:

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

示例2:

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

两个数组总和之差的一半一定是交换的一对数值之差。假设数组1的总和为sum1, 数组2的总和为sum2。那么如果sum1+b1-a1 = sum2+a1-b1, 则sum1-sum2=(a1-b1)*2,a1-b1 = (sum1-sum2)/2。如果两数组中有元素存在这种关系，则返回对应值。

代码如下：

vector<int> findSwapValues(vector<int>& array1, vector<int>& array2) 
    {
        int sum1=0, sum2=0;
        unordered_set<int> s;
        for(auto &c1:array1) sum1+=c1;
        for(auto &c2:array2) {sum2+=c2; s.insert(c2);}
        if(abs(sum2-sum1)%2==1) return{};
        else
        {
            for(int i=0;i<array1.size();i++)
            if(s.find(array1[i]+(sum2-sum1)/2) != s.end())
                return {array1[i], array1[i]+(sum2-sum1)/2};
        }
        return {};
    }