349.两个数组的交集
给定两个数组,编写一个函数来计算它们的交集。
输出结果中的每个元素都唯一。可以不考虑输出结果的顺序。
输入: nums1 = [1,2,2,1], nums2 = [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]:
set1 = set(nums1)
ans = []
for x in nums2:
if x in set1:
ans += x,
set1.remove(x)
return ans
# or return list(set(nums1) & set(nums2))