350.两个数组的交集2
题目
给定两个数组,编写一个函数来计算它们的交集。
示例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映射的字典,找出它的key值是否在nums2中,并比对nums2映射的字典中该key对应的value,选出最小的value作为最终交集的元素的重复次数。
代码:
class Solution(object):
def intersect(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
if not nums1:return []
if not nums2:return []
hashmap1={}
hashmap2={}
for i in range(len(nums1)):
hashmap1[nums1[i]]=hashmap1.get(nums1[i],0)+1
for i in range(len(nums2)):
hashmap2[nums2[i]]=hashmap2.get(nums2[i],0)+1
result=[]
min_len=0
for k,v in hashmap1.items():
if k in nums2:
min_len=min(v,hashmap2[k])
for i in range(min_len):
result.append(k)
min_len=0
return result
测试用例
def intersect(nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
if not nums1:return []
if not nums2:return []
hashmap1={}
hashmap2={}
for i in range(len(nums1)):
hashmap1[nums1[i]]=hashmap1.get(nums1[i],0)+1
for i in range(len(nums2)):
hashmap2[nums2[i]]=hashmap2.get(nums2[i],0)+1
result=[]
min_len=0
for k,v in hashmap1.items():
if k in nums2:
min_len=min(v,hashmap2[k])
for i in range(min_len):
result.append(k)
min_len=0
return result
nums1 = [4,9,5]
nums2 = [9,4,9,8,4]
print(intersect(nums1, nums2))
我看了一下官方答案:
class Solution(object):
def intersect(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
if not nums1 or not nums2: return []
hashmap1=dict.fromkeys(nums1,0)
hashmap2=dict.fromkeys(nums2,0)
for num in nums1:
hashmap1[num]+=1
for num in nums2:
if hashmap1.get(num) and hashmap1[num]>0:
hashmap2[num]+=1
hashmap1[num]-=1
res=[]
for k,v in hashmap2.items():
for _ in range(v):
res.append(k)
return res
它初始化了字典个数,首先在第二个for循环那里就把重复项放入字典,并借助hashmap1的value来统计重复元素的频次,放入hashmap2一次,value-1。这样,比我的方法更加清晰省时。