[刷题] 350 Intersection of Two Arrays

要求

• 给定两个数组nums，求两个数组交集
• 输出结果与元素在两个数组中出现的次数一致
• 不考虑输出结果的顺序

举例

• nums1=[1,2,2,1]
• nums2=[2,2]
• 结果：[2,2]

思路

• 使用map，记录nums1中元素及其出现次数
• 遍历nums2，将重复元素放入结果，同时将该元素出现次数减1

实现

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