滴,二月卡,希望这个月题目别像上个月一样orz
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 delta = (sumA - sumB) / 2;
unordered_set<int> rec(A.begin(), A.end());
vector<int> ans;
for (auto a : B)
{
int x = a + delta;
if (rec.count(x))
{
ans = vector<int>{x, a};
break;
}
}
return ans;
}
};
说几个我感觉要注意的地方叭,首先两者的差不要加绝对值!!!如果加了绝对值还要比较两者和的大小,比较麻烦,还有就是最后ans里面元素的顺序的问题,题目要求的是Alice and Bob,所以如果你遍历Bob的糖果,那么得到的结果应该放在前面,反之亦然,因为最后的结果肯定是和遍历相反的,因为想达到两者糖果一样肯定要交换,这点我想了好久,或许我比较笨叭,希望我说明白了