python(leetcode)-350两个数组的交集

时间:2023-03-09 15:22:37
python(leetcode)-350两个数组的交集
给定两个数组,编写一个函数来计算它们的交集。

示例 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 的大小比 nums2 小很多,哪种方法更优?
如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?

看完问题描述 发现看似简单 但是其实有一点点的难 难点在于如果2个数组中存在重复数字,则交集也必须重复

先说下 自己当时的想法

没有想到什么简单方法 只能使用蛮力法 通过两层循环 每一个数字与另外一个数组中的数字对比。并且加一个判断,如果相同则在之后的对比时不对比此下标的值

上代码(通过-用时110ms)

 class Solution:

     def intersect2(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
result = [] for i in nums1:
for j in nums2:
if i == j:
result.append(i) #插入交集
nums2.remove(j) #插入过的值不再出现
break
return result
if __name__=="__main__":
nums1=[1,2,2,1]
nums2=[1,2]
s=Solution()
print(s.intersect(nums1,nums2))

这个是笔者想到的方法。使用蛮力法,比较耗时,需要110ms。

然后上评论区大佬的代码(特别赏心悦目 5行代码解决问题)

 class Solution(object):
def intersect(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
inter = set(nums1) & set(nums2) #inter为两个list的相同元素
l = []
for i in inter:
l += [i] * min(nums1.count(i), nums2.count(i)) #相同元素的共同出现次数
return l

这个方法比较难想,是一个很有趣的解法

思想:通过set的唯一性找出两list的唯一交集,再利用min方法找到交集的共同出现次数

前面方法的效率较低 介绍一种方法使用字典进行查找

思想:通过key,value进行存储,key为值的个数,value为值。 先把nums1存放入字典中。遍历nums2中的值对比字典,相同则数量减一并放入list中,最后返回list

 class Solution:
#利用字典实现交集
def intersect(self,nums1,nums2):
record={}
result=[]
for i in nums1:
# 注意python3.x用contains()方法取代has_key()
if(record.__contains__(i)):
record[i]+=1 #将字典的key值数+1
else:
record[i]=1 #将i加入字典中 key为1
for j in nums2:
if(record.__contains__(j) and record[j]>0):
record[j]-=1 #i的key值减1
result.append(j)
return result
if __name__=="__main__":
s=Solution()
num1=[1,2,3,4,4,1]
num2=[4,2,3]
print(s.intersect(num1,num2))

值得注意的是:python 3.x的版本将has_key()方法换为_contains_()

会发现通过时间为48ms 效率有所提升