350. 两个数组的交集 II(python)

最新推荐文章于 2022-12-08 10:08:56 发布

misite_J

最新推荐文章于 2022-12-08 10:08:56 发布

阅读量342

点赞数

分类专栏： Leecode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/misite_J/article/details/107324719

版权

Leecode 专栏收录该内容

2 篇文章

订阅专栏

本文详细解析了LeetCode上350题“两个数组的交集II”的算法实现，包括遍历匹配和哈希表优化两种方法，旨在帮助读者理解如何计算两个数组的交集，特别关注元素出现次数的最小值一致性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述

350. 两个数组的交集 II 给定两个数组，编写一个函数来计算它们的交集。

示例 1：

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

示例 2:

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

说明：

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

问题分析

直接的想法是：遍历数组1，查看是否含于数组2，如果是的话，删除数组2中的该元素，直至数组1循环完毕。接下来考虑是将长数组(n)还是短数组(m)作为数组1，结论是：遍历长数组，匹配短数组。理由如下：虽然两种情况的最坏最好时间复杂度都是 $O(m^2)$ 、 $O(n^2)$ ，但是遍历长数组过程中，匹配短数组有这样一个条件：当短数组空时，打破循环，因此，相对而言遍历长数组时间复杂度更优。实验验证也是如此。
使用哈希表结构进一步优化时间复杂度。

实现代码

# 遍历匹配
class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        if len(nums1) > len(nums2):
            res = self.find(nums1, nums2)
        else:
            res = self.find(nums2, nums1)
        
        return res

    def find(self, longnums, shortnums):
        res = []
        for i in range(len(longnums)):
            if shortnums==[]:
                break
            if longnums[i] in shortnums:
                res.append(longnums[i])
                shortnums.remove(longnums[i])

        return res

# 哈希表结构
class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        res_dict = {}
        res = []
        for i in nums1:
            if i in res_dict:
                res_dict[i] += 1
            else:
                res_dict[i] = 1
        for j in nums2:
            if j not in res_dict:
                continue
            else:
                if res_dict[j] == 0:
                    continue
                else:
                    res.append(j)
                    res_dict[j] -= 1
    
        return res