leetcode_350 两个数组的交集 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 的元素存储在磁盘上，内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？

核心代码

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        if len(nums1) > len(nums2):
            nums1,nums2 = nums2,nums1
        res = []
        for i in nums1:
            if i in nums2:
                res.append(i)
                nums2.remove(i)
        return res

另一解法

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        res_dict = {}
        res = []
        for i in nums1:
            if i in res_dict:
                res_dict[i] += 1
            else:
                res_dict[i] = 1
        for j in nums2:
            if j not in res_dict:
                continue
            else:
                if res_dict[j] == 0:
                    continue
                else:
                    res.append(j)
                    res_dict[j] -= 1
        return res

解题思路：第一种解法：我们使用remove方法，从第二个列表中删除对应元素，防止干扰下次的判断，比较简单。第二种解法：我们使用哈希表的方式解决此题，我们先统计一个列表中的字符及字符的个数，最终能扫描第二个列表，最终得到交集。