leetcode-350:两个数组的交集 II(python中Counter的用法,海象运算符:=)

简介: leetcode-350:两个数组的交集 II(python中Counter的用法,海象运算符:=)

题目

题目链接

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

示例 1:

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

示例 2:

输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]

解题:

方法一:(lc官方的解法)哈希表

参考链接

由于同一个数字在两个数组中都可能出现多次,因此需要用哈希表存储每个数字出现的次数。对于一个数字,其在交集中出现的次数等于该数字在两个数组中出现次数的最小值。

首先遍历第一个数组,并在哈希表中记录第一个数组中的每个数字以及对应出现的次数,然后遍历第二个数组,对于第二个数组中的每个数字,如果在哈希表中存在这个数字,则将该数字添加到答案,并减少哈希表中该数字出现的次数。

为了降低空间复杂度,首先遍历较短的数组并在哈希表中记录每个数字以及对应出现的次数,然后遍历较长的数组得到交集。

image.png


这是官方的代码,但是有些啰嗦

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        if len(nums1) > len(nums2):
            return self.intersect(nums2, nums1)
        m = collections.Counter()
        for num in nums1:
            m[num] += 1
        intersection = list()
        for num in nums2:
            if (count := m.get(num, 0)) > 0:# := 有点啰嗦,后续根本没有用到count
                intersection.append(num)
                m[num] -= 1
                if m[num] == 0:   # 这部分也很啰嗦,因为前面的if里面,m[num]一定是大于0的
                    m.pop(num)
        return intersection

删除掉那些啰嗦的部分后

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        if len(nums1)>len(nums2):
            return self.intersect(nums2,nums1)
        m = Counter()
        for num in nums1:
            m[num]+=1
        interaction = []
        for num in nums2:
            if m.get(num,0)>0:# m.get(num,0)这个0意思是如果不存在,默认给0
                m[num]-=1
                interaction.append(num)
        return interaction

如果不用Counter,用普通的字典,那么

m = collections.Counter()
        for num in nums1:
            m[num] += 1

就要改为

dict1 = {}
        for one in nums1:
            if dict1.get(one)==None:
                dict1[one]=1
            else:
                dict1[one]+=1

方法二:巧用Counter

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        num1 = collections.Counter(nums1)
        num2 = collections.Counter(nums2)
        num = num1 & num2
        return list(num.elements())

方法三:排序

扩展:求两数组的并集

利用上述的思想,可以写出求两数组的并集

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        m1 = Counter(nums1)
        m2 = Counter(nums2)
        for num in nums2:
            if m1[num]<m2[num]:
                m1[num] = m2[num]
        return list(m1.elements())

Counter 与 普通的字典的差异

Counter中文意思是计数器,也就是我们常用于统计的一种数据类型,在使用Counter之后可以让我们的代码更加简单易读

Counter实际上是个字典的一种类。

对于访问用索引不存在的键的值,Conter会返回0,而普通的字典会报错。普通的字典要通过.get()方法,条件语句来计数。

from collections import Counter
m = Counter()
print(m["a"])
>>0
m = dict{}
print(m["a"])
>>报错:KeyError: 'a'
if m.get("a") == None:
  ....
else:

Counter详细解释的链接

Counter类型数据中的&操作:

from collections import Counter
list1 = [1,1,2,2,3]
list2 = [1,1]
m1 = Counter(list1)
m2 = Counter(list2)
print(m1&m2)
>>Counter({1: 2})

Counter的&运算键对应的值,取小的那个。

海象运算符:=

(Python 3.8 的新特性)

:=运算符可以在使用的同时进行赋值,使得代码更加简练

比如:

进行判断,同时给age赋值

if (age:= 20) > 18:
    print("已经成年了")

循环的条件

while (p := input("Enter the password: ")) != "youpassword":
   continue

海象运算符链接

相关文章
|
存储 Java 数据处理
(numpy)Python做数据处理必备框架!(一):认识numpy;从概念层面开始学习ndarray数组:形状、数组转置、数值范围、矩阵...
Numpy是什么? numpy是Python中科学计算的基础包。 它是一个Python库,提供多维数组对象、各种派生对象(例如掩码数组和矩阵)以及用于对数组进行快速操作的各种方法,包括数学、逻辑、形状操作、排序、选择、I/0 、离散傅里叶变换、基本线性代数、基本统计运算、随机模拟等等。 Numpy能做什么? numpy的部分功能如下: ndarray,一个具有矢量算术运算和复杂广播能力的快速且节省空间的多维数组 用于对整组数据进行快速运算的标准数学函数(无需编写循环)。 用于读写磁盘数据的工具以及用于操作内存映射文件的工具。 线性代数、随机数生成以及傅里叶变换功能。 用于集成由C、C++
883 1
|
机器学习/深度学习 并行计算 大数据
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧2
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧
841 10
|
索引 Python
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧1
【Python篇】NumPy完整指南(上篇):掌握数组、矩阵与高效计算的核心技巧
732 4
|
存储 Python
Python多个set中的交集
Python多个set中的交集
625 1
|
存储 自然语言处理 数据处理
使用Python计算多个集合的交集详解
使用Python计算多个集合的交集详解
999 1
|
存储 数据处理 索引
如何删除 Python 数组中的值?
【8月更文挑战第29天】
936 9
|
C++
【LeetCode 12】349.两个数组的交集
【LeetCode 12】349.两个数组的交集
211 0
|
Unix Shell Linux
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
本文提供了几个Linux shell脚本编程问题的解决方案,包括转置文件内容、统计词频、验证有效电话号码和提取文件的第十行,每个问题都给出了至少一种实现方法。
637 6
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
|
Python
【Leetcode刷题Python】剑指 Offer 32 - III. 从上到下打印二叉树 III
本文介绍了两种Python实现方法,用于按照之字形顺序打印二叉树的层次遍历结果,实现了在奇数层正序、偶数层反序打印节点的功能。
314 6
|
搜索推荐 索引 Python
【Leetcode刷题Python】牛客. 数组中未出现的最小正整数
本文介绍了牛客网题目"数组中未出现的最小正整数"的解法,提供了一种满足O(n)时间复杂度和O(1)空间复杂度要求的原地排序算法,并给出了Python实现代码。
657 2

推荐镜像

更多