题目描述:
- 爱丽丝和鲍勃有不同大小的糖果棒:A[i] 是爱丽丝拥有的第 i 根糖果棒的大小,B[j] 是鲍勃拥有的第 j 根糖果棒的大小。
因为他们是朋友,所以他们想交换一根糖果棒,这样交换后,他们都有相同的糖果总量。(一个人拥有的糖果总量是他们拥有的糖果棒大小的总和。)
返回一个整数数组 ans,其中 ans[0] 是爱丽丝必须交换的糖果棒的大小,ans[1] 是 Bob 必须交换的糖果棒的大小。
如果有多个答案,你可以返回其中任何一个。保证答案存在。
示例 :
输入:A = [1,1], B = [2,2]
输出:[1,2]
分析:
-
最后是两个人要实现一样多,简单分析即可;设爱丽丝糖果和是sumA,鲍勃糖果和是sumB,{x,y}是交换数,依题意我们可以得到这样一个等式:
sumA - x + y = sumB - y + x
得到x和y的关系:
x = (sumA-sumB+2*y)/2 -
下面我们可以上代码了,
- 先求和
//1.求和
int sumA = Arrays.stream(A).sum();
int sumB = Arrays.stream(B).sum();
- 把爱丽丝所有糖果放入map集合中,
//2.制造map集合,并将A全部元素放入
Map<Integer,Integer> map = new HashMap<>();
for (int i = 0; i < A.length; i++) {
map.put(A[i],null);
}
- 遍历鲍勃的糖果,利用哈希表去寻找是否存在糖果数,判断其是否在A中出现,即可得出结果
//3. 遍历B
for (int j : B) {
int re = (sumA-sumB+2*j)/2;
if(map.containsKey(re)){
return new int[]{re,j};
}
}
代码:
public int[] fairCandySwap(int[] A, int[] B) {
//1.求和
int sumA = Arrays.stream(A).sum();
int sumB = Arrays.stream(B).sum();
//2.制造map集合,并将A全部元素放入
Map<Integer,Integer> map = new HashMap<>();
for (int i = 0; i < A.length; i++) {
map.put(A[i],null);
}
//3. 遍历B
for (int j : B) {
int re = (sumA-sumB+2*j)/2;
if(map.containsKey(re)){
return new int[]{re,j};
}
}
return null;
}
总结:
- 此题算是简单题,只要列出公式,然后实现即可,要知道使用哈希表来求解。