题目

哈希表

推导出映射关系使用哈希表求解：
假设A的总和为sumA, B的总和为sumB，交换元素x， y后A、B总和相等，即 s u m A − x + y = s u m B − y + x , 化 简 得 ： x = y + ( s u m A − s u m B ) / 2 sumA - x + y = sumB - y + x,化简得： x = y + (sumA - sumB) / 2 sumA−x+y=sumB−y+x,化简得：x=y+(sumA−sumB)/2,将A中元素加入哈希表，遍历B中元素，如果符合上述映射关系，则找到了交换得一对元素。

class Solution {
    public int[] fairCandySwap(int[] A, int[] B) {
        Set<Integer> set = new HashSet<>();
        int sumA = Arrays.stream(A).sum();
        int sumB = Arrays.stream(B).sum();
        int delta = (sumA - sumB) / 2;
        for (int a : A) {
            set.add(a);
        }
        for (int b : B) {
            if (set.contains(b + delta)) {
                return new int[]{b + delta, b};
            }
        }
        return new int[0];
    }
}