美文网首页
数组—— 两个数组的交集 II

数组—— 两个数组的交集 II

作者: CoeusZ | 来源:发表于2019-01-31 10:12 被阅读0次

    题目

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

    示例 1:

    输入: nums1 = [1,2,2,1], nums2 = [2,2]
    输出: [2,2]
    

    示例 2:

    输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
    输出: [4,9]
    

    说明:

    输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
    我们可以不考虑输出结果的顺序。

    解题思路

    一开始,我采取了最粗暴的方法,遍历一个列表,拿出每一个元素去判断该元素是否在另一个列表中,如果在,就把这个元素拿出来放在一个地方,同时在另一个列表中删除该元素。最后所有拿出来的元素就是交集。
    显然,这很慢。

    开始想第二种方法,把这两个列表都转成字典,值是每个元素出现的次数,然后再来遍历某个字典,依次拿出key去判断是否在另一个字段中,如果存在,再取出2个字典中,值小的那个一个并乘以值作为个数,用一个列表接收。
    这个,比上面快了一点点。

    别人提交的最快的答案:
    其实我的第二种思路已经接近了,但是没有必要将两个列表都转换成字典,只要将一个列表转换成字典,然后遍历另一个列表,这样就将复杂度降低了一个N。

    答案(一)

    class Solution(object):
        def intersect(self, nums1, nums2):
            """
            :type nums1: List[int]
            :type nums2: List[int]
            :rtype: List[int]
            """
            result = []
            for i in nums1:
                if i in nums2:
                    nums2.remove(i)
                    result.append(i)
    
            return result
    

    答案(二)

    from collections import Counter
    
    class Solution(object):
        def intersect(self, nums1, nums2):
            """
            :type nums1: List[int]
            :type nums2: List[int]
            :rtype: List[int]
            """
            result = []
            nums1_dict = Counter(nums1)
            nums2_dict = Counter(nums2)
            for i in nums1_dict:
                if i in nums2_dict:
                    result += [i] * min(nums1_dict[i], nums2_dict[i])
                    
            return result
    

    答案(三)

    class Solution(object):
        def intersect(self, nums1, nums2):
            """
            :type nums1: List[int]
            :type nums2: List[int]
            :rtype: List[int]
            """
            dictnums1 = dict()
            ans = []
            for num in nums1:
                dictnums1[num] = dictnums1[num] + 1 if num in dictnums1 else 1
            for num in nums2:
                if num in dictnums1 and dictnums1[num]>=1:
                    ans.append(num)
                    dictnums1[num] -=1
    

    相关文章

      网友评论

          本文标题:数组—— 两个数组的交集 II

          本文链接:https://www.haomeiwen.com/subject/bycksqtx.html