Leetcode 面试题 16.21.交换和
1 题目描述(Leetcode题目链接)
给定两个整数数组,请交换一对数值(每个数组中取一个数值),使得两个数组所有元素的和相等。
返回一个数组,第一个元素是第一个数组中要交换的元素,第二个元素是第二个数组中要交换的元素。若有多个答案,返回任意一个均可。若无满足条件的数值,返回空数组。
输入: array1 = [4, 1, 2, 1, 1, 2], array2 = [3, 6, 3, 3]
输出: [1, 3]
输入: array1 = [1, 2, 3], array2 = [4, 5, 6]
输出: []
提示:
- 1 <= array1.length, array2.length <= 100000
2 题解
变成集合,然后挨个找就可以了。
class Solution:
def findSwapValues(self, array1: List[int], array2: List[int]) -> List[int]:
s1, s2 = sum(array1), sum(array2)
arr1, arr2 = set(array1), set(array2)
for i in arr1:
j = i + (s2-s1)//2
if 2*(j-i) == s2 - s1 and j in arr2:
return [i, j]
return []