公平的糖果交换

发布时间 2023-06-27 19:36:12作者: 网抑云黑胶SVIP用户

爱丽丝和鲍勃拥有不同总数量的糖果。给你两个数组 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]

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

class Solution {
    public int[] fairCandySwap(int[] aliceSizes, int[] bobSizes) {
        //数组总值
        int sumA = Arrays.stream(aliceSizes).sum();
        int sumB = Arrays.stream(bobSizes).sum();
        //式子sumA−x+y=sumB+x−y =>x=y+(sumA−sumB)/2
        int delta = (sumA-sumB)/2;
        HashSet<Integer> set = new HashSet<>();
        for(int aliceSize : aliceSizes){
            set.add(aliceSize);
        }
        int[] resutlt = new int[2];
        for(int bobSize : bobSizes){
            if(set.contains(bobSize+delta)){
                resutlt[0]=bobSize+delta;
                resutlt[1]=bobSize;
                break;
            }
        }
        return resutlt;
    }
}