力扣链接:888. 公平的糖果交换 - 力扣(LeetCode)
爱丽丝和鲍勃拥有不同总数量的糖果。给你两个数组 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)
{
vector<int> res;
std::sort(A.begin(), A.end()); // 排序
std::sort(B.begin(), B.end()); // 排序
int sum_a = 0, sum_b = 0; // 二者各自的总糖果数
for (int i = 0; i < A.size(); i++) sum_a += A[i];
for (int i = 0; i < B.size(); i++) sum_b += B[i];
int diff = sum_a - sum_b; // 二者差值
int i = 0, j = 0; // 双指针分别指向两个数组的起始位置
// 找到并交换两个差值为diff/2的糖果棒,即可使二者糖果数相等
while (i < A.size() && j < B.size())
{
if (A[i] - B[j] > diff / 2) // 差值太小,减数增大
j++;
else if (A[i] - B[j] < diff / 2) // 差值太小,被减数增大
i++;
else // 找到需要交换的两个数
{
res.push_back(A[i]);
res.push_back(B[j]);
break;
}
}
return res;
}
};
此方法首先对两个数组排序,求得两数组之和的差值,所以我们只要找到ali和bob数组中相差为此差值的一组值,就是题解。此方法采用排序后的数组逐步逼近两数之和的差值。
方法二:
class Solution {
public:
vector<int> fairCandySwap(vector<int>& aliceSizes, vector<int>& bobSizes) {
int aliceSum = std::accumulate(aliceSizes.begin(), aliceSizes.end(), 0);
int bobSum = std::accumulate(bobSizes.begin(), bobSizes.end(), 0);
int dif = (aliceSum + bobSum)/2 - aliceSum;
for (int aliceCandy : aliceSizes) {
int bobCandy = aliceCandy + dif;
if (bobCandies.count(bobCandy)) {
return {aliceCandy, bobCandy};
}
}
return {};
}
};
方法二简化了很多,主要思路是求出两数组之和,用两数组之和除二减去ali的和,意思是求出ali还差多少可以达到相等,同时也是bob相对于相等而言多的总数。根据题意,我们需要把这种整体差值放到一对特定数组值的里面。我们把这种差值加上ali数组的每一个值,并且在bob数组中查找是否有相同的数字。
本题中需要注意这种解题方法,同时这里面有一个问题就是Alice的总数总是比bob少,所以bob每次都要把自己的糖果给alice,请大家端正态度,不做舔狗。