力扣 888. 公平的糖果棒交换 哈希 枚举

https://leetcode-cn.com/problems/fair-candy-swap/

思路：看数据范围知道暴力枚举是不行滴，先转化问题，不妨设$sumA={\sum }_{i=0}^n{A}_i$，$sum=sumA+{\sum }_{i=0}^m{B}_i$，那么此时我们枚举$i$，假设爱丽丝需要交换的糖果数是$A[i]$，那么显然要存在一个$j$，使得下式成立：
$$sumA-A[i]+B[j]=sum/2$$
由此可知：
$$B[j]=sum/2-sumA+A[i]$$
也就是说对于每个$i$，只需要判断数组$B$中是否存在$sum/2-sumA+A[i]$这个数即可，那么有什么数据结构可以快速做到这一点呢？当然是哈希表了。

class Solution {
public:
    vector<int> fairCandySwap(vector<int>& A, vector<int>& B) {
        map<int,bool> m;
        int sum=0,sumA=0,n=A.size();
        for(int i=0;i<n;i++)
            sum+=A[i];
        sumA=sum;
        n=B.size();
        for(int i=0;i<n;i++)
            sum+=B[i],m[B[i]]=1;
        sum>>=1;
        auto it=m.end();
        n=A.size();
        for(int i=0;i<n;i++)
        {
            if(m.find(sum-sumA+A[i])!=it)
                return {A[i],sum-sumA+A[i]};
        }
        return {0,0};
    }
};