LeetCode 350. 两个数组的交集 II（C++、python）

最新推荐文章于 2020-11-02 09:59:57 发布

原创最新推荐文章于 2020-11-02 09:59:57 发布 · 722 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

713 篇文章

订阅专栏

本文介绍了一种计算两个数组交集的方法，包括C++和Python两种实现方式。通过使用映射和列表，算法能够准确地计算出两个数组中共同元素的出现次数，并保持一致性。

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

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

示例 1:

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

示例 2:

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

说明：

输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。

我们可以不考虑输出结果的顺序。

C++

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) 
    {
        vector<int> res;
        map<int,int> temp;
        for(int i=0;i<nums1.size();i++)
        {
            temp[nums1[i]]++;
        }
        for(int i=0;i<nums2.size();i++)
        {
            if(temp[nums2[i]]>0)
            {
                res.push_back(nums2[i]);
                temp[nums2[i]]--;
            }
        }
        return res;
    }
};

python（方法一）

class Solution:
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        temp=[]
        for i in range(0,len(nums2)):
            if nums2[i] in nums1:
                temp.append(nums2[i])
                nums1.remove(nums2[i])
                i -= 1
        return temp

python（方法二）

class Solution:
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        dic=dict()
        res=[]
        for i in range(0,len(nums1)):
            if nums1[i] not in dic:
                dic[nums1[i]] = 1
            else:
                dic[nums1[i]] += 1
        for i in range(0,len(nums2)):
            if nums2[i] in dic:
                if(dic[nums2[i]]>0):
                    res.append(nums2[i])
                    dic[nums2[i]]-=1
        return res