Leetcode 面試題 16.21.交換和

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 []