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]}