888. 公平的糖果交换

爱丽丝和鲍勃拥有不同总数量的糖果。给你两个数组 aliceSizes 和 bobSizes ，aliceSizes[i] 是爱丽丝拥有的第 i 盒糖果中的糖果数量，bobSizes[j] 是鲍勃拥有的第 j 盒糖果中的糖果数量。

两人想要互相交换一盒糖果，这样在交换之后，他们就可以拥有相同总数量的糖果。一个人拥有的糖果总数量是他们每盒糖果数量的总和。

返回一个整数数组 answer，其中 answer[0] 是爱丽丝必须交换的糖果盒中的糖果的数目，answer[1] 是鲍勃必须交换的糖果盒中的糖果的数目。如果存在多个答案，你可以返回其中 任何一个 。题目测试用例保证存在与输入对应的答案。

class Solution {
public:
    vector<int> fairCandySwap(vector<int>& A, vector<int>& B) {
        int sumA = accumulate(A.begin(),A.end(),0);
        int sumB = accumulate(B.begin(),B.end(),0);
        int num = abs(sumA - sumB) / 2;
        if(sumA > sumB){
            for(const auto n : A){
                if(binary_search(B.begin(),B.end(),n - num))    return {n, n - num};
            }
        }
        for(const auto n : B){
            if(binary_search(A.begin(),A.end(),n - num))    return {n - num, n};
        }
        return {-1,-1};
    }
};

gpt优化

class Solution {
public:
    vector<int> fairCandySwap(vector<int>& A, vector<int>& B) {
        int sumA = accumulate(A.begin(), A.end(), 0);
        int sumB = accumulate(B.begin(), B.end(), 0);

        int diff = (sumA - sumB) / 2;

        unordered_set<int> setB(B.begin(), B.end());
        for (const auto& a : A) {
            int target = a - diff;
            if (setB.count(target)) {
                return {a, target};
            }
        }

        return {}; // 如果无法找到答案，返回空向量表示题目条件不满足
    }
};