【LeetCode】公平的糖果交换 java

题目

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

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

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

示例 1：

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

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

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

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

题解

// 暴力
    public int[] fairCandySwap(int[] aliceSizes, int[] bobSizes) {
        int sumA = Arrays.stream(aliceSizes).sum();
        int sumB = Arrays.stream(bobSizes).sum();
        int temp = sumA-sumB;
        int step = temp / 2;
        for (int aliceSize : aliceSizes) {
            for (int bobSize : bobSizes) {
                if (aliceSize - step == bobSize && bobSize + step == aliceSize) {
                    return new int[]{aliceSize, bobSize};
                }
            }
        }
        return new int[2];
    }


    // 哈希表
    public int[] fairCandySwap1(int[] aliceSizes, int[] bobSizes) {
        int sumA = Arrays.stream(aliceSizes).sum();
        int sumB = Arrays.stream(bobSizes).sum();
        Map<Integer,Integer> map = new HashMap<>();
        int temp = sumA-sumB;
        int step = temp / 2;
        for (int aliceSize : aliceSizes) {
            map.put(aliceSize -step, aliceSize);
        }
        for (int bobSize : bobSizes) {
            if (map.containsKey(bobSize)) {
                return new int[]{map.get(bobSize), bobSize};
            }
        }
        return new int[2];
    }

    // 排序 + 二分法
    public int[] fairCandySwap2(int[] aliceSizes, int[] bobSizes) {
        int sumA = Arrays.stream(aliceSizes).sum();
        int sumB = Arrays.stream(bobSizes).sum();

        int temp = sumA-sumB;
        int step = temp / 2;

        Arrays.sort(aliceSizes);
        for (int bobSize : bobSizes) {
            if(check(bobSize + step, aliceSizes)) {
                return new int[]{bobSize + step, bobSize};
            }
        }
        return new int[2];
    }

    private boolean check(int target, int[] aliceSizes) {
        int left = 0;
        int right = aliceSizes.length - 1;
        while (left <= right ) {
            int mid = (right - left)/2 + left;
            if (aliceSizes[mid] == target) {
                return true;
            } else if (aliceSizes[mid] > target) {
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }
        return false;
    }