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

海象运算符链接

相关文章
|
9月前
|
存储 Java 数据处理
(numpy)Python做数据处理必备框架!(一):认识numpy;从概念层面开始学习ndarray数组:形状、数组转置、数值范围、矩阵...
Numpy是什么? numpy是Python中科学计算的基础包。 它是一个Python库,提供多维数组对象、各种派生对象(例如掩码数组和矩阵)以及用于对数组进行快速操作的各种方法,包括数学、逻辑、形状操作、排序、选择、I/0 、离散傅里叶变换、基本线性代数、基本统计运算、随机模拟等等。 Numpy能做什么? numpy的部分功能如下: ndarray,一个具有矢量算术运算和复杂广播能力的快速且节省空间的多维数组 用于对整组数据进行快速运算的标准数学函数(无需编写循环)。 用于读写磁盘数据的工具以及用于操作内存映射文件的工具。 线性代数、随机数生成以及傅里叶变换功能。 用于集成由C、C++
717 1
|
9月前
|
Python
Python编程:运算符详解
本文全面详解Python各类运算符,涵盖算术、比较、逻辑、赋值、位、身份、成员运算符及优先级规则,结合实例代码与运行结果,助你深入掌握Python运算符的使用方法与应用场景。
528 3
|
人工智能 Python
Python 中的 `and`, `or`, `not` 运算符
本文介绍了 Python 中的逻辑运算符 `and`、`or` 和 `not` 的基本用法及其特性。这些运算符主要用于布尔运算,特别是在条件判断和循环中非常有用。文章详细解释了每个运算符的功能,例如 `and` 检查所有表达式是否为真,`or` 检查是否有任意一个表达式为真,`not` 用于取反。此外,还提到这些运算符支持短路特性,并可应用于非布尔值场景。掌握这些运算符有助于编写更高效、简洁的代码。
941 11
|
Go
【LeetCode 热题100】DP 实战进阶:最长递增子序列、乘积最大子数组、分割等和子集(力扣300 / 152/ 416 )(Go语言版)
本文深入解析三道经典的动态规划问题:**最长递增子序列(LIS)**、**乘积最大子数组** 和 **分割等和子集**。 - **300. LIS** 通过 `dp[i]` 表示以第 `i` 个元素结尾的最长递增子序列长度,支持 O(n²) 动态规划与 O(n log n) 的二分优化。 - **152. 乘积最大子数组** 利用正负数特性,同时维护最大值与最小值的状态转移方程。 - **416. 分割等和子集** 转化为 0-1 背包问题,通过布尔型 DP 实现子集和判断。 总结对比了三题的状态定义与解法技巧,并延伸至相关变种问题,助你掌握动态规划的核心思想与灵活应用!
518 1
|
Python
【10月更文挑战第7天】「Mac上学Python 14」基础篇8 - 运算符详解
本篇将详细介绍Python中的运算符,包括数学运算、关系运算、逻辑运算、赋值运算和成员运算等内容。同时会说明运算符的优先级和运算顺序,帮助用户理解和掌握Python中的运算符使用规则。
307 3
【10月更文挑战第7天】「Mac上学Python 14」基础篇8 - 运算符详解
|
知识图谱 Python
Python入门:4.Python中的运算符
Python是一间强大而且便捷的编程语言,支持多种类型的运算符。在Python中,运算符被分为算术运算符、赋值运算符、复合赋值运算符、比较运算符和逻辑运算符等。本文将从基础到进阶进行分析,并通过一个综合案例展示其实际应用。
|
Python
Python运算符优先级
Python运算符优先级。
289 2
|
Python
Python成员运算符
Python成员运算符
202 2
|
Python
Python赋值运算符
Python赋值运算符。
191 2
|
Python
Python比较运算符
Python比较运算符。
237 2

推荐镜像

更多