Python描述 LeetCode 888. 公平的糖果交换

  大家好，我是亓官劼（qí guān jié ），在【亓官劼】公众号、GitHub、B站等平台分享一些技术博文，主要包括前端开发、python后端开发、小程序开发、数据结构与算法、docker、Linux常用运维、NLP等相关技术博文，时光荏苒，未来可期，加油~

  如果喜欢博主的文章可以关注博主的个人公众号【亓官劼】(qí guān jié)，里面的文章更全更新更快。如果有需要找博主的话可以在公众号后台留言，我会尽快回复消息.

本文原创为【亓官劼】（qí guān jié ），请大家支持原创，部分平台一直在恶意盗取博主的文章！！！ 全部文章请关注微信公众号【亓官劼】。

题目

爱丽丝和鲍勃拥有不同总数量的糖果。给你两个数组 ​

​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]      

提示：

• ​

  ​1 <= aliceSizes.length, bobSizes.length <= 104​

  ​
• ​

  ​1 <= aliceSizes[i], bobSizes[j] <= 105​

  ​
• 爱丽丝和鲍勃的糖果总数量不同。
• 题目数据保证对于给定的输入至少存在一个有效答案。

Python描述一

class Solution:
    def fairCandySwap(self, aliceSizes: List[int], bobSizes: List[int]) -> List[int]:
        aliceSizes.sort()
        bobSizes.sort()
        a_sum = sum(aliceSizes)
        b_sum = sum(bobSizes)
        for item in aliceSizes:
            l,r = 0,len(bobSizes)-1
            while l <= r:
                mid = l + r + 1 >> 1
                if a_sum - item + bobSizes[mid] == b_sum - bobSizes[mid] + item:
                    return [item,bobSizes[mid]]
                if a_sum - item + bobSizes[mid] < b_sum - bobSizes[mid] + item:
                    l = mid + 1
                else:
                    r = mid - 1      

Python描述二

class Solution:
    def fairCandySwap(self, aliceSizes: List[int], bobSizes: List[int]) -> List[int]:
        aliceSizes.sort(),bobSizes.sort()
        a_sum,b_sum = sum(aliceSizes),sum(bobSizes)
        b_dict = {}
        for item in bobSizes:
            b_dict[item] = b_dict.get(item,0)+1
        for item in aliceSizes:
            tmp = (2*item+b_sum-a_sum) // 2
            if b_dict.get(tmp,False):
                return [item,tmp]