开启掘金成长之旅！这是我参与「掘金日新计划 · 2 月更文挑战」的第 6 天，点击查看活动详情
算法题目：LeetCode-888
爱丽丝和鲍勃拥有不同总数量的糖果。给你两个数组 aliceSizes 和 bobSizes ，aliceSizes[i] 是爱丽丝拥有的第 i 盒糖果中的糖果数量，bobSizes[j] 是鲍勃拥有的第 j 盒糖果中的糖果数量。
两人想要互相交换一盒糖果，这样在交换之后，他们就可以拥有相同总数量的糖果。一个人拥有的糖果总数量是他们每盒糖果数量的总和。
返回一个整数数组 answer，其中 answer[0] 是爱丽丝必须交换的糖果盒中的糖果的数目，answer[1] 是鲍勃必须交换的糖果盒中的糖果的数目。如果存在多个答案，你可以返回其中 任何一个 。题目测试用例保证存在与输入对应的答案。
示例 1：
输入： aliceSizes = [1,1], bobSizes = [2,2]
输出： [1,2]
示例 2：
输入： aliceSizes = [1,2], bobSizes = [2,3]
输出： [1,2]
示例 3：
输入： aliceSizes = [2], bobSizes = [1,3]
输出： [2,3]
示例 4：
输入： aliceSizes = [1,2,5], bobSizes = [2,4]
输出： [5,4]
提示：
1 <= aliceSizes[i], bobSizes[j] <= 105- 爱丽丝和鲍勃的糖果总数量不同。
- 题目数据保证对于给定的输入至少存在一个有效答案。
解题分析
由题意分析先算出两个数组的总和sumA，sumB
- 然后算出一个和之差的一半:(sumB-sumA)/2，这个数就是需要的数
- 在集合一中，有一个数，加上这个需要的数，能在B中找到，这两个数就是答案
- 因此之前在算B的和的时候，顺便将B中元素存入一个集合
算法代码
public int[] fairCandySwap(int[] A, int[] B) {
int sumA = 0;
int sumB = 0;
Set < Integer > set = new HashSet < > ();
//获取两个数组的总和
for (int a: A) {
sumA += a;
}
for (int b: B) {
sumB += b;
set.add(b);
}
//得到总和的差的一半，就是A集合中一个数要加上这个need数，会是B中存在这个数
int need = (sumB - sumA) / 2;
for (int a: A) {
int curNeed = a + need;
if (set.contains(curNeed)) {
return new int[] {
a, curNeed
};
}
}
return null;
}
代码执行
分析复杂度
- 时间复杂度：
- 空间复杂度：
在掘金(JUEJIN) 一起分享知识, Keep Learning!
