350.intersection-of-two-arrays-ii

时间:2022-04-02 22:07:15

给定两个数组,写一个方法来计算它们的交集。

例如:
给定 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