LeetCode刷题（11）～ 两个数组的交集 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]

解答

提交代码

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        //假定num1 短
        if(nums1.size()>nums2.size())
        return intersect(nums2,nums1);
         
         unordered_map<int,int> a;
         for(int i=0;i<nums2.size();++i)
         ++a[nums2[i]];

         vector<int> b;
         for(int i=0;i<nums1.size();++i)
         {
             if(a.count(nums1[i]))
             {
                 b.push_back(nums1[i]);
                 --a[nums1[i]];
                 if(a[nums1[i]]==0)
                 a.erase(nums1[i]);
             }
         }

         return b;
        
    }
};

运行结果

思路
        哈希表！ 详解

题目来源

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/intersection-of-two-arrays-ii