888. 公平的糖果棒交换（哈希表）
公平的糖果棒交换
爱丽丝和鲍勃有不同大小的糖果棒：A[i] 是爱丽丝拥有的第 i 根糖果棒的大小，B[j] 是鲍勃拥有的第 j 根糖果棒的大小。因为他们是朋友，所以他们想交换一根糖果棒，这样交换后，他们都有相同的糖果总量。（一个人拥有的糖果总量是他们拥有的糖果棒大小的总和。）
返回一个整数数组 ans，其中 ans[0] 是爱丽丝必须交换的糖果棒的大小，ans[1] 是 Bob 必须交换的糖果棒的大小。如果有多个答案，你可以返回其中任何一个。保证答案存在。
示例 1：
输入：A = [1,1], B = [2,2]
输出：[1,2]
示例 2：
输入：A = [1,2], B = [2,3]
输出：[1,2]
示例 3：
输入：A = [2], B = [1,3]
输出：[2,3]
示例 4：
输入：A = [1,2,5], B = [2,4]
输出：[5,4]
解题思路
(1) 设 Alice 的糖果总数为 sumA，Bob 的糖果总数为 sumB。设答案为 {x, y}，即 Alice 交换的糖果棒的大小为 x，Bob 交换糖果棒的大小为 y，则有等式 sumA - x + y = sumB - y + x 通过移项得到 x = y + (sumA - sumB)/2，所以对于 y，只要找到对应的 x 即可。
(2) 首先将 A 中的所有元素都放到哈希表中(因为哈希表查找起来更快)，然后遍历数组 B，根据公式求出对应 x 的值，若该值在哈希表中存在则直接返回 {x, y}
方法一：使用哈希表来存储数组 A 中的元素，遍历数组 B 得到 y 后，从哈希表中找对应的 x 的值。
方法二：遍历数组 B 得到 y 后，直接遍历数组 A 找对应的 x 的值。
代码
方法一
class Solution {
public int[] fairCandySwap(int[] A, int[] B) {
//直接对A和B数组中的值求和
int sumA = Arrays.stream(A).sum();
int sumB = Arrays.stream(B).sum();
int count = (sumA - sumB) / 2;
//定义哈希表
//HashSet中只能存储不能重复的对象，如果重复了后面的内容会覆盖前面的
Set<Integer> res = new HashSet<Integer>();
for (int num : A) {
res.add(num);
}
int[] ans = new int[2];
for (int y : B) {
int x = y + count;
//判断是否有某个元素，直接使用set.contains()
if (res.contains(x)) {
ans[0] = x;
ans[1] = y;
break; //因为只需要返回其中一个答案就行(选择第一个返回即可)
}
}
return ans;
}
}
时间复杂度：遍历了数组 A 和 B，所以复杂度为 O(m + n)
空间复杂度：开辟了新的空间，其中 HashSet 的大小与数组 A 的长度相同，且定义了数组 ans 长度为 2，复杂度为 O(n + 2)，但由于常数归到 n 中，所以最后的复杂度为 O(n)
方法二
class Solution {
public int[] fairCandySwap(int[] A, int[] B) {
int n = A.length;
int m = B.length;
int sumA = 0, sumB = 0;
for (int i=0;i<n;i++) {
sumA += A[i];
}
for (int i=0;i<m;i++) {
sumB += B[i];
}
int count = (sumA - sumB) / 2;
int[] ans = new int[2];
for (int i=0;i<m;i++) {
int x = B[i] + count;
for (int j=0;j<n;j++) {
if (x == A[j]) {
ans[0] = x;
ans[1] = B[i];
break;
}
}
}
return ans;
}
}
方法二的速度比方法一的执行用时要慢的多，还是用哈希表更好一些！