LeetCode 两个数组的交集 II

传送门

题目描述

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

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]

示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明：
输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。

进阶:

如果给定的数组已经排好序呢？你将如何优化你的算法？
如果 nums1 的大小比 nums2 小很多，哪种方法更优？
如果 nums2 的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

解法一

• 时间复杂度：O(n²)

遍历nums1和nums2，发现相同的就加到结果集中。

def intersect(self, nums1, nums2):
    result = []
    for i in nums1:
        for index, j in enumerate(nums2):
            if i == j:
                result.append(i)
                nums2[index] = None
                break
    return result

解法二

• 时间复杂度：O( nlog(n) + n )

先给nums1和nums2排序，然后遍历nums1和nums2，如果

• nums1[i] > nums2[j]：j++
• nums1[i] < nums2[j]： i++
• nums1[i] == nums2[j]：i++, j++

def intersect(self, nums1, nums2):
    nums1.sort()
    nums2.sort()
    result, i, j = [], 0, 0
    len1, len2 = len(nums1), len(nums2)

    while i < len1 and j < len2:
        a, b = nums1[i], nums2[j]
        if a == b:
            result.append(a)
            i, j = i + 1, j + 1
        elif a > b:
            j += 1
        else:
            i += 1
    return result

解法三

• 时间复杂度：O(n)

先统计nums1中各个数字出现的次数，存到d中，然后遍历nums2，如果当前遍历数字在d中存在，则加到结果集中，并且d[n] -= 1，如果d[n]为0，则把n从d中移除。

def intersect(self, nums1, nums2):
    d = {}
    for i in nums:
        d[i] = d.get(i, 0) + 1
    result = []
    for n in nums2:
        if n in d:
            result.append(n)
            if d[n] == 1:
                d.pop(n)
            else:
                d[n] -= 1
    return result

解法四

利用了python的collections模块中Counter类，分别给nums1和nums2中各个数字统计出现次数，然后用& 操作符取交集，通过Counter类的elements方法即可取得结果集。

def intersect(self, nums1, nums2):
    return list((collections.Counter(nums1) & collections.Counter(nums2)).elements())