leetcode面试题16.21.交换和

面试题16.21.交换和

给定两个整数数组，请交换一对数值（每个数组中取一个数值），使得两个数组所有元素的和相等。

返回一个数组，第一个元素是第一个数组中要交换的元素，第二个元素是第二个数组中要交换的元素。

若有多个答案，返回任意一个均可。若无满足条件的数值，返回空数组。

示例:

输入: array1 = [4, 1, 2, 1, 1, 2], array2 = [3, 6, 3, 3]

输出: [1, 3]

示例:

输入: array1 = [1, 2, 3], array2 = [4, 5, 6]

输出: []

def func(array1, array2):
    diff = sum(array1) - sum(array2)
    if diff % 2 == 1:
        return []
    else:
        diff >>= 1
        array2 = set(array2)
        for i in array1:
            if i - diff in array2:
                return [i, i - diff]
    return []


print(func(array1 = [4, 1, 2, 1, 1, 2], array2 = [3, 6, 3, 3]))
``