888. 公平的糖果交换

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第5天，点击查看活动详情

题目 leetcode.cn/

• 爱丽丝和鲍勃拥有不同总数量的糖果。给你两个数组 aliceSizes 和 bobSizes ，aliceSizes[i] 是爱丽丝拥有的第 i 盒糖果中的糖果数量，bobSizes[j] 是鲍勃拥有的第 j 盒糖果中的糖果数量。
• 两人想要互相交换一盒糖果，这样在交换之后，他们就可以拥有相同总数量的糖果。一个人拥有的糖果总数量是他们每盒糖果数量的总和。
• 返回一个整数数组 answer，其中 answer[0] 是爱丽丝必须交换的糖果盒中的糖果的数目，answer[1] 是鲍勃必须交换的糖果盒中的糖果的数目。如果存在多个答案，你可以返回其中 任何一个 。题目测试用例保证存在与输入对应的答案。

示例

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

提示

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

代码

function fairCandySwap(aliceSizes: number[], bobSizes: number[]): number[] {
    let result = [];
    let alice = aliceSizes.reduce((next, pre) => {
        return next + pre;
    }, 0);
    let bob = bobSizes.reduce((next, pre) => {
        return next + pre;
    }, 0);
    let target = 0;
    if(alice > bob){
        target = (alice - bob) / 2;
        for(let i = 0; i < aliceSizes.length; i++){
            for(let j = 0; j < bobSizes.length; j++){
                if(aliceSizes[i] - bobSizes[j] === target){
                    result.push(aliceSizes[i], bobSizes[j]);
                    return result;
                }
            }
        }
    } else {
        target= (bob - alice) / 2;
        for(let i = 0; i < aliceSizes.length; i++){
            for(let j = 0; j < bobSizes.length; j++){
                if(bobSizes[j] - aliceSizes[i] === target){
                    result.push(aliceSizes[i], bobSizes[j]);
                    return result;
                }
            }
        }
    }
};

思路

• 要想找到他们要交换的糖果，首先要知道他们距离糖果相等时各差了或者多了几个
  • 先将两个人原有的糖果数量求和
  • 然后判断是爱丽丝的糖果多还是鲍勃的糖果多
  • 如果爱丽丝的糖果多，那么在后面的交换过程中，爱丽丝拿出的糖果肯定比鲍勃多，并且是她多于鲍勃糖果数量的一半
  • 如果鲍勃的糖果多，那么在后面的交换过程中，鲍勃拿出的糖果是他多于爱丽丝糖果数量的一半
• 那么就很简单了，遍历他们的糖果，找到两个盒子里的糖果差值等于他们多的一方的一半就行
  • 因为只要输出一个答案，所以如果找到了就立即退出循环返回就好了