LeetCode：888. Fair Candy Swap - Python

问题描述：

888. 公平的糖果交换
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))

声明： 总结学习，大神可以略过哦，有问题可以批评指正哦。