两个数组的交集 II
leetcode 350
给定两个数组,编写一个函数来计算它们的交集。
示例 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 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
哈希表
可以分为两步:
- 统计词频。
先在字典dic
记录一个数组中的存在的数字和对应出现的次数。 - 验证是否存在。
遍历第二个数组,检查数字在dic
中是否存在,如果存在,将该数字添加到答案并减少dic
中的计数。
注意:比较大小。检查数组的大小并对较小的数组进行哈希映射是一个小细节,当其中一个数组较大时,会减少内存的使用。
from collections import Counter
def intersect(nums1, nums2):
if len(nums1) > len(nums2):
nums1, nums2 = nums2, nums1
dic = Counter(nums1) # 统计词频
# for i in nums1:
# dic[i] = dic.get(i, 0) + 1
res = []
for i in nums2:
if dic.get(i): # 结果要么为 None, 要么为 0,要么为 > 0 的数。
dic[i] -= 1
res.append(i)
return res
其中,引入了 c