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;
    }
}

方法二的速度比方法一的执行用时要慢的多，还是用哈希表更好一些！