两个数组的交集 II:Python 语法糖 Counter

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

哈希表

可以分为两步:

  1. 统计词频。
    先在字典 dic 记录一个数组中的存在的数字和对应出现的次数。
  2. 验证是否存在。
    遍历第二个数组,检查数字在 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

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值