问题描述:
Alice和Bob 有不同大小的糖果棒:
A[i]
是爱丽丝拥有的第
i
个糖果棒的大小,而
B[j]
是鲍勃拥有的第j个糖果棒的大小。
由于他们是朋友,他们想交换一个糖果,以便交换后,他们都有相同的糖果总量。 (一个人拥有的糖果总量是他们拥有的糖果大小的总和。)
返回一个整数数组ans,其中ans[0]
是Alice必须交换糖果棒大小,ans[1]
是Bob必须交换糖果棒大小。
如果有多个答案,您可以返回其中任何一个。 保证答案存在。
示例1:
输入: A = [1,1], B = [2,2]
输出: [1,2]
注意:
1 <= A.length <= 10000
1 <= B.length <= 10000
1 <= A[i] <= 100000
1 <= B[i] <= 100000
- 保证Alice和Bob的糖果总量不同。
- 保证有答案。
问题分析:
题目说的很明白了,一定有答案。我们设:Alice和Bob 分别拿出的是x
、y
,所以有:
sum(A) - x + y = sum(B) - y + x
y = (sum(B) - sum(A))/2 + x
Python3实现:
# @Time :2018/08/12
# @Author :LiuYinxing
class Solution:
def fairCandySwap(self, A, B):
totalA, totalB, setB = sum(A), sum(B), set(B)
for v in A:
if (totalB - totalA)//2 + v in setB:
return [v, (totalB - totalA)//2 + v]
if __name__ == '__main__':
A, B = [1, 2, 5], [2, 4]
solu = Solution()
print(solu.fairCandySwap(A, B))
声明: 总结学习,大神可以略过哦,有问题可以批评指正哦。