程序员面试金典 16.21

Sum Swap：给定两个数组，从每个数组中取出一个元素，使得交换这两个元素后，两数组和相等。

最简单的方法就是暴力枚举，时间复杂度为O(AB)，其中A和B分别表示数组大小。写代码之前看了下测试用例的范围，数组大小上限为100000，估计肯定会超时的，果不其然。

优化只能从内层循环入手，尝试使用map或者set代替array2的遍历，复杂度降为O(AlogB)。

class Solution {
public:
    vector<int> findSwapValues(vector<int>& array1, vector<int>& array2) {
        int sum1 = accumulate(array1.begin(), array1.end(), 0);
        int sum2 = accumulate(array2.begin(), array2.end(), 0);
        if((sum1 + sum2) % 2 != 0) return {};
        int half = (sum1 + sum2) / 2;
        set<int> set2;
        for_each(array2.begin(), array2.end(), [&set2](const int i){ set2.insert(i); });
        for(size_t i = 0; i < array1.size(); i++)
        {
            int target = half - sum1 + array1[i];
            if(set2.find(target) != set2.end()){
                return { array1[i], target };
            }
        }
        return {};
    }
};

如果两个数组都是有序的，则可以同时顺序遍历找出结果，时间复杂度为O(AlogA + BlogB)。

class Solution {
public:
    vector<int> findSwapValues(vector<int>& array1, vector<int>& array2) {
        int sum1 = accumulate(array1.begin(), array1.end(), 0);
        int sum2 = accumulate(array2.begin(), array2.end(), 0);
        if((sum1 + sum2) % 2 != 0) return {};
        int half = (sum1 + sum2) / 2;
        sort(array1.begin(), array1.end());
        sort(array2.begin(), array2.end());
        size_t i = 0, j = 0;
        while(i < array1.size() && j < array2.size()){
            int ChangedSum = sum1 - array1[i] + array2[j];
            if(ChangedSum == half){
                return { array1[i], array2[j] };
            }
            else if(ChangedSum > half){
                i++;
            }
            else{
                j++;
            }
        }
        return {};
    }
};