给定两个数组,写一个方法来计算它们的交集。
例如:
给定 nums1 = [1, 2, 2, 1]
, nums2 = [2, 2]
, 返回 [2, 2]
.
注意:
- 输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
- 我们可以不考虑输出结果的顺序。
跟进:
- 如果给定的数组已经排好序呢?你将如何优化你的算法?
- 如果 nums1 的大小比 nums2 小很多,哪种方法更优?
- 如果nums2的元素存储在磁盘上,内存是有限的,你不能一次加载所有的元素到内存中,你该怎么办?
解题思路:
第一反应是在349那道题的基础上,把del那行删了就行,还是太年轻了。。。如果输入是[1]. [1,1]就不行了,还是得先将两个数组排序然后,同时遍历两个数组,并设置i,j两个变量。
解题代码:
ans=[] nums1.sort() nums2.sort() i=j=0 while i<len(nums1) and j<len(nums2): if nums1[i]<nums2[j]: i+=1 elif nums2[j]<nums1[i]: j+=1 else: ans.append(nums2[j]) i+=1 j+=1 return ans