888. 公平的糖果交换

Posted on | In
Words count in article: 412 | Reading time ≈ 1

题目

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

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

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

示例1：

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

示例2：

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

示例3：

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

示例4：

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

提示：

  1. 1 <= aliceSizes.length, bobSizes.length <= 10^4
  2. 1 <= aliceSizes[i], bobSizes[j] <= 10^5
  3. 爱丽丝和鲍勃的糖果总数量不同。
  4. 题目数据保证对于给定的输入至少存在一个有效答案。

解法

解法一：

java

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
public int[] fairCandySwap(int[] aliceSizes, int[] bobSizes) {
            int sumA = Arrays.stream(aliceSizes).sum();
            int sumB = Arrays.stream(bobSizes).sum();
            int delta = (sumA - sumB) / 2;
            Set<Integer> rec = new HashSet<>();
            for (int num : aliceSizes) {
                rec.add(num);
            }
            int[] ans = new int[2];
            for (int y : bobSizes) {
                int x = y + delta;
                if (rec.contains(x)) {
                    ans[0] = x;
                    ans[1] = y;
                    break;
                }
            }
            return ans;
        }
0%