面试题 16.21. 交换和

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

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

示例:

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

示例:

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

class Solution {
public:
    vector<int> findSwapValues(vector<int>& array1, vector<int>& array2) {
        vector<int> res;
        set<int> a, b;
        int sum_1 = 0, sum_2 = 0;
        for(int i = 0; i < array1.size(); ++i){
            sum_1 += array1[i];
            a.insert(array1[i]);
        }
        for(int i = 0; i < array2.size(); ++i){
            sum_2 += array2[i];
            b.insert(array2[i]);
        }
        if((sum_1 - sum_2) % 2 != 0){
            return {};
        }
        int difference = (sum_1 - sum_2) / 2;
        // sum_1 - difference = sum_2
        /*
            [1, 2, 3]
            [4, 5, 6]
        */
        for(int i : a){
            if(b.count(i - difference)){
                return {i, i - difference};
            }
        }
        return {};
    }
};