题目
给定两个数组,编写一个函数来计算它们的交集。
示例 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官方的解法)哈希表
由于同一个数字在两个数组中都可能出现多次,因此需要用哈希表存储每个数字出现的次数。对于一个数字,其在交集中出现的次数等于该数字在两个数组中出现次数的最小值。
首先遍历第一个数组,并在哈希表中记录第一个数组中的每个数字以及对应出现的次数,然后遍历第二个数组,对于第二个数组中的每个数字,如果在哈希表中存在这个数字,则将该数字添加到答案,并减少哈希表中该数字出现的次数。
为了降低空间复杂度,首先遍历较短的数组并在哈希表中记录每个数字以及对应出现的次数,然后遍历较长的数组得到交集。
这是官方的代码,但是有些啰嗦
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类型数据中的&操作:
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