力扣 888. 公平的糖果交换

题目

爱丽丝和鲍勃拥有不同总数量的糖果。给你两个数组 aliceSizes 和 bobSizes ，aliceSizes[i] 是爱丽丝拥有的第 i 盒糖果中的糖果数量，bobSizes[j] 是鲍勃拥有的第 j 盒糖果中的糖果数量。

两人想要互相交换一盒糖果，这样在交换之后，他们就可以拥有相同总数量的糖果。一个人拥有的糖果总数量是他们每盒糖果数量的总和。

返回一个整数数组 answer，其中 answer[0] 是爱丽丝必须交换的糖果盒中的糖果的数目，answer[1] 是鲍勃必须交换的糖果盒中的糖果的数目。如果存在多个答案，你可以返回其中 任何一个 。题目测试用例保证存在与输入对应的答案。

示例

输入：aliceSizes = [1,1], bobSizes = [2,2]
输出：[1,2]

输入：aliceSizes = [1,2], bobSizes = [2,3]
输出：[1,2]

输入：aliceSizes = [2], bobSizes = [1,3]
输出：[2,3]

输入：aliceSizes = [1,2,5], bobSizes = [2,4]
输出：[5,4]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/fair-candy-swap
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：暴力模拟

Java实现

class Solution {
    public int[] fairCandySwap(int[] aliceSizes, int[] bobSizes) {
        int sumA = 0, sumB = 0;

        for (int a : aliceSizes) sumA += a;
        for (int b : bobSizes) sumB += b;
        
        for (int a : aliceSizes) {
            for (int b : bobSizes) {
                if (sumA - a + b == sumB - b + a) return new int[] {a, b};
            }
        }
        return new int[]{};
    }
}

方法2：数学+哈希表

Java实现

class Solution {
    public int[] fairCandySwap(int[] aliceSizes, int[] bobSizes) {
        int sumA = 0, sumB = 0;
        Map<Integer, Integer> map = new HashMap<>();

        for (int a : aliceSizes) {
            sumA += a;
            map.put(a, map.getOrDefault(a, 0) + 1);
        }
        for (int b : bobSizes) sumB += b;

        int cha = (sumA - sumB) / 2;

        for (int y : bobSizes) {
            int x = y + cha;
            if (map.containsKey(x)) return new int[]{x, y};
        }
        return new int[]{};
    }
}