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

2019-01-31  本文已影响0人  CoeusZ

题目

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

示例 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
上一篇 下一篇

猜你喜欢

热点阅读