leetcode 349. 两个数组的交集（简单）

leetcode 349. 两个数组的交集（简单）

给定两个数组，编写一个函数来计算它们的交集。

示例 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]:
        ans=[]
        for i in nums1:
            if i in nums2 and i not in ans:
                ans.append(i)
        return ans

可以发现这个方法在处理数组中有很多重复元素时，会执行大量的重复操作，所以可以采用集合的方法。

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        set1=set(nums1)
        set2=set(nums2)
        ans=[]
        if len(set1)>len(set2):
            set1,set2=set2,set1
        return [x for x in set1 if x in set2]

使用这个方法可以避免很多的重复操作，但是在官方题解中这个方法的时间复杂度是O(m+n)，这里我还有点疑问。

后来我发现运来in操作在处理数组时时间复杂度是O(n)，所以用数组解决这个问题的时间复杂度是O(mn)。
但是in操作处理字典及集合时的时间复杂度是O(1)，所以总的时间复杂度其实是一开始将数组放入元素的时间，即O(m+n)。