给定两个数组,编写一个函数来计算它们的交集。
示例 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)
1 class Solution:
2
3 def intersect2(self, nums1, nums2):
4 """
5 :type nums1: List[int]
6 :type nums2: List[int]
7 :rtype: List[int]
8 """
9 result = []
10
11 for i in nums1:
12 for j in nums2:
13 if i == j:
14 result.append(i) #插入交集
15 nums2.remove(j) #插入过的值不再出现
16 break
17 return result
18 if __name__=="__main__":
19 nums1=[1,2,2,1]
20 nums2=[1,2]
21 s=Solution()
22 print(s.intersect(nums1,nums2))
这个是笔者想到的方法。使用蛮力法,比较耗时,需要110ms。
然后上评论区大佬的代码(特别赏心悦目 5行代码解决问题)
1 class Solution(object):
2 def intersect(self, nums1, nums2):
3 """
4 :type nums1: List[int]
5 :type nums2: List[int]
6 :rtype: List[int]
7 """
8 inter = set(nums1) & set(nums2) #inter为两个list的相同元素
9 l = []
10 for i in inter:
11 l += [i] * min(nums1.count(i), nums2.count(i)) #相同元素的共同出现次数
12 return l
这个方法比较难想,是一个很有趣的解法
思想:通过set的唯一性找出两list的唯一交集,再利用min方法找到交集的共同出现次数
前面方法的效率较低 介绍一种方法使用字典进行查找
思想:通过key,value进行存储,key为值的个数,value为值。 先把nums1存放入字典中。遍历nums2中的值对比字典,相同则数量减一并放入list中,最后返回list
1 class Solution:
2 #利用字典实现交集
3 def intersect(self,nums1,nums2):
4 record={}
5 result=[]
6 for i in nums1:
7 # 注意python3.x用contains()方法取代has_key()
8 if(record.__contains__(i)):
9 record[i]+=1 #将字典的key值数+1
10 else:
11 record[i]=1 #将i加入字典中 key为1
12 for j in nums2:
13 if(record.__contains__(j) and record[j]>0):
14 record[j]-=1 #i的key值减1
15 result.append(j)
16 return result
17 if __name__=="__main__":
18 s=Solution()
19 num1=[1,2,3,4,4,1]
20 num2=[4,2,3]
21 print(s.intersect(num1,num2))
值得注意的是:python 3.x的版本将has_key()方法换为_contains_()
会发现通过时间为48ms 效率有所提升
不太相信弯道超车,更欣赏仰望星空与脚踏实地