leetcode349.两个数组的交集

197 阅读1分钟

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))