Leetcode每日一题：888. 公平的糖果棒交换

最新推荐文章于 2023-03-28 22:03:43 发布

原创最新推荐文章于 2023-03-28 22:03:43 发布 · 173 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #数据结构 #python

leetcode练习专栏收录该内容

47 篇文章

订阅专栏

本文介绍了一个有趣的问题——如何让两个朋友通过交换一根糖果棒达到拥有相同糖果总量的目标。通过两种不同的算法实现，一种是直接遍历查找，另一种是利用哈希表提高查找效率。

问题描述

爱丽丝和鲍勃有不同大小的糖果棒：A[i] 是爱丽丝拥有的第 i 根糖果棒的大小，B[j] 是鲍勃拥有的第 j 根糖果棒的大小。

因为他们是朋友，所以他们想交换一根糖果棒，这样交换后，他们都有相同的糖果总量。（一个人拥有的糖果总量是他们拥有的糖果棒大小的总和。）

返回一个整数数组 ans，其中 ans[0] 是爱丽丝必须交换的糖果棒的大小，ans[1] 是 Bob 必须交换的糖果棒的大小。

如果有多个答案，你可以返回其中任何一个。保证答案存在。

示例 1：
输入：A = [1,1], B = [2,2]
输出：[1,2]
示例 2：
输入：A = [1,2], B = [2,3]
输出：[1,2]
示例 3：
输入：A = [2], B = [1,3]
输出：[2,3]
示例 4：
输入：A = [1,2,5], B = [2,4]
输出：[5,4]

提示：

1 <= A.length <= 10000
1 <= B.length <= 10000
1 <= A[i] <= 100000
1 <= B[i] <= 100000
保证爱丽丝与鲍勃的糖果总量不同。答案肯定存在。

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

思路分析及代码实现

我一开始是这样做的，先求出A与平均数的差值，然后遍历转为集合的A，将每个元素都加上差值看看是否在转为集合的B中，结果虽然通过了，
然时间慢的可怜

class Solution:
    def fairCandySwap(self, A: List[int], B: List[int]) -> List[int]:
        sub = (sum(A) + sum(B)) // 2 - sum(A)
        for i in set(A):
        	if i + sub in set(B):
        		return [i, i+sub]

在这里插入图片描述

于是我用了哈希表的做法，将集合A填入哈希表中，判断B中元素加上差值是否在哈希表A，这样大大提高了效率

class Solution:
    def fairCandySwap(self, A: List[int], B: List[int]) -> List[int]:
        sub = (sum(A) + sum(B)) // 2 - sum(B)
        hashdic = {}
        for i in set(A):
        	if i not in hashdic:
        		hashdic[i] = 0
        for i in set(B):
            if i + sub in hashdic.keys():
                return [i+sub,i]