350. Intersection of Two Arrays II

问题描述

解决方案

我的方法

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int> mi1,mi2;
        vector<int> vi;
        for(auto &num:nums1)
        {
            mi1[num]++;
        }
        for(auto &num:nums2)
        {
            mi2[num]++;
        }
        for(unordered_map<int,int>::iterator iter=mi1.begin();iter!=mi1.end();++iter)
        {
            int i=0;
            if(mi2.count(iter->first)>0)
            {
                i=min(mi1[iter->first],mi2[iter->first]);
                while(i--)
                {
                    vi.push_back(iter->first);
                }
            }
        }
        return vi;
    }
};

别人的

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