两个数组的交集(350)

题目:两个数组的交集(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]

 

原文地址:https://www.cnblogs.com/yan-2010/p/14155436.html