【Leetcode 888 】 888. 公平的糖果交换 —— 纯数学题

63 阅读1分钟

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

纯数学题

//纯数学题
function fairCandySwap(aliceSizes: number[], bobSizes: number[]): number[] {
  // 假设答案 res的结果为 [x,y]
  //则满足 sliceSum - x + y = bobSum - y + x
  //简化为 sliceSum - bobSum + 2y = 2x
  //简化为 (sliceSum - bobSum) / 2 + y = x
  //调换位置  x = (sliceSum - bobSum) / 2 + y

  //求和
  let sliceSum = aliceSizes.reduce((pre, cur) => (pre += cur), 0);
  let bobSum = bobSizes.reduce((pre, cur) => (pre += cur), 0);
  const aliceSet = new Set(aliceSizes);
  let res: [number, number] = [0, 0];
  for (const sweet of bobSizes) {
    //求出x
    const x = (sliceSum - bobSum) / 2 + sweet;
    //如果 aliceSet 存在这个x,则表明此交换是公平的
    if (aliceSet.has(x)) {
      res = [x, sweet];
      break;
    }
  }
  return res;
}