LeetCode 350. 两个数组的交集 II(C++、python)

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

示例 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 intersect(vector& nums1, vector& nums2) 
    {
        vector res;
        map temp;
        for(int i=0;i0)
            {
                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

 

你可能感兴趣的:(LeetCode)