Python描述 LeetCode 888. 公平的糖果交换 大家好,我是亓官劼(qí guān jié ),在【亓官劼】公众号、GitHub、B站等平台分享一些技术博文,主要包括前端开发、python后端开发、小程序
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]