Class Solution

Array/FairCandySwap.py:71–82

"""

class Solution(object):
    def fairCandySwap(self, A, B):
        """
        :type A: List[int]
        :type B: List[int]
        :rtype: List[int]
        """
        x = (sum(A) - sum(B)) // 2
        b = set(B)        
        for i in A:
            if i - x in b:
                return (i, i-x)