题目:两个数组的交集(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 的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
解题思路
首先遍历第一个数组,记录下第一个数组每个元素和其出现的次数,然后遍历第二个数组,如果第二个数组中出现了该元素,那么把该元素添加到结果中,并把该元素元素在第一个数组中出现的次数减1
这里记录元素出现的次数,很容易想到用python 内置模块 Counter
python 代码:
class Solution(object): def intersect(self, nums1, nums2): """ :type nums1: List[int] :type nums2: List[int] :rtype: List[int] """ if len(nums1)>len(nums2): nums1,nums2= nums2,nums1 c=Counter(nums1) result=[] for value in nums2: if c[value]>0: c[value]-=1 result.append(value) return result
解法二:
class Solution(object): def intersect(self, nums1, nums2): """ :type nums1: List[int] :type nums2: List[int] :rtype: List[int] """ nums1.sort() nums2.sort() i=j=k=0 result=[] while((i<len(nums1)) and (j<len(nums2))): if nums1[i]<nums2[j]: i+=1 elif nums1[i]>nums2[j]: j+=1 else : nums2[k]=nums2[j] i+=1 j+=1 k+=1 return nums2[:k]