題目:
兩個數組的交集:給定兩個數組,編寫一個函數來計算它們的交集。
示例 1:
輸入: nums1 = [1,2,2,1], nums2 = [2,2]
輸出: [2]
示例 2:
輸入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
輸出: [9,4]
說明:
輸出結果中的每個元素一定是唯一的。
我們可以不考慮輸出結果的順序。
思路:
看到要求不重復結果使用哈希表。
程序:
class Solution:
def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
if not nums1 and nums2:
return []
if nums1 and not nums2:
return []
if not nums1 and not nums2:
return []
result = set()
for index in range(len(nums1)):
if nums1[index] in nums2:
result.add(nums1[index])
return result
