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

海象运算符链接

相关文章
|
1月前
|
算法
Leetcode 初级算法 --- 数组篇
Leetcode 初级算法 --- 数组篇
38 0
|
1月前
|
Python
【10月更文挑战第7天】「Mac上学Python 14」基础篇8 - 运算符详解
本篇将详细介绍Python中的运算符,包括数学运算、关系运算、逻辑运算、赋值运算和成员运算等内容。同时会说明运算符的优先级和运算顺序,帮助用户理解和掌握Python中的运算符使用规则。
37 3
【10月更文挑战第7天】「Mac上学Python 14」基础篇8 - 运算符详解
|
1月前
|
自然语言处理 Python
【python从入门到精通】-- 第三战:输入输出 运算符
【python从入门到精通】-- 第三战:输入输出 运算符
65 0
【python从入门到精通】-- 第三战:输入输出 运算符
|
1月前
【LeetCode-每日一题】 删除排序数组中的重复项
【LeetCode-每日一题】 删除排序数组中的重复项
19 4
|
1月前
|
存储 Java 编译器
Python学习三:学习python的 变量命名规则,算数、比较、逻辑、赋值运算符,输入与输出。
这篇文章是关于Python编程语言中变量命名规则、基本数据类型、算数运算符、比较运算符、逻辑运算符、赋值运算符以及格式化输出与输入的详细教程。
18 0
Python学习三:学习python的 变量命名规则,算数、比较、逻辑、赋值运算符,输入与输出。
|
1月前
|
索引
Leetcode第三十三题(搜索旋转排序数组)
这篇文章介绍了解决LeetCode第33题“搜索旋转排序数组”的方法,该问题要求在旋转过的升序数组中找到给定目标值的索引,如果存在则返回索引,否则返回-1,文章提供了一个时间复杂度为O(logn)的二分搜索算法实现。
18 0
Leetcode第三十三题(搜索旋转排序数组)
|
1月前
|
存储 Python
Python多个set中的交集
Python多个set中的交集
|
1月前
|
存储 自然语言处理 数据处理
使用Python计算多个集合的交集详解
使用Python计算多个集合的交集详解
38 1
|
1月前
|
算法 C++
Leetcode第53题(最大子数组和)
这篇文章介绍了LeetCode第53题“最大子数组和”的动态规划解法,提供了详细的状态转移方程和C++代码实现,并讨论了其他算法如贪心、分治、改进动态规划和分块累计法。
56 0
|
1月前
|
C++
【LeetCode 12】349.两个数组的交集
【LeetCode 12】349.两个数组的交集
16 0