888. Fair Candy Swap

class Solution {
public:
    vector<int> fairCandySwap(vector<int>& A, vector<int>& B) {
        int sumA=0,sumB=0;
        vector<int> result;
        for(int i=0;i<A.size();i++){
            sumA+=A[i];
        }
        for(int i=0;i<B.size();i++){
            sumB+=B[i];
        }
        for(int i=0;i<A.size();i++){
            for(int j=0;j<B.size();j++){
                if(sumB-B[j]+A[i]==sumA-A[i]+B[j]){
                    result.push_back(A[i]);
                    result.push_back(B[j]);
                    return result;
                }
            }
        }
    }
};

My solution

每个人总数固定，减去自己的加上别人的，个数相等就好啦。

 

Better Solution

因为最后两个人个数相等，想到平均数，最开始算出平均数，只要数目到了这个数就好了，不过也是两层循环

 

Attention

vector return的时候可以用大括号表示{A[i],B[j]}