公平的糖果交换

/*
 * 思路：
 * 1. 计算爱丽丝和鲍勃各自的糖果总数。
 * 2. 计算需要交换的糖果数目差 diff = (aliceSum - bobSum) / 2。
 * 3. 用一个集合来存储爱丽丝所有糖果的数目。
 * 4. 使用 Java Stream 遍历鲍勃的糖果数目，检查是否存在爱丽丝的糖果数目满足 aliceCandy - bobCandy = diff。
 * 5. 找到符合条件的糖果数目，返回结果。
 */
import java.util.*;
import java.util.stream.*;

public class CandySwapStream {
    public int[] fairCandySwap(int[] aliceSizes, int[] bobSizes) {
        int aliceSum = Arrays.stream(aliceSizes).sum();
        int bobSum = Arrays.stream(bobSizes).sum();
        int diff = (aliceSum - bobSum) / 2;

        Set<Integer> aliceSet = Arrays.stream(aliceSizes).boxed().collect(Collectors.toSet());

        return Arrays.stream(bobSizes)
                .filter(candy -> aliceSet.contains(candy + diff))
                .mapToObj(candy -> new int[]{candy + diff, candy})
                .findFirst()
                .orElse(new int[0]);  // 实际上这里不会到达，题目保证有答案
    }
}