两个数组的交集 II(C++)

248 阅读1分钟

两个数组的交集 II

给你两个整数数组 nums1 和 nums2 ,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序。

示例 1:

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

示例 2:

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

提示:

  • 1 <= nums1.length, nums2.length <= 1000
  • 0 <= nums1[i], nums2[i] <= 1000

我的思路: 这个题也不是很难,考点就是map的使用,相对于两个数组的交集I,区别在于不去重。 我的代码:

class Solution {
public:
// 这个题比较特别
// 需要的是把所有重复的元素找出来
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
    // 空间复杂度优化的小技巧
        if (nums1.size() > nums2.size()) return intersect(nums2, nums1);
        unordered_map<int, int> mp;
        // 先统计nums1中的数字
        for (int num:nums1) mp[num] ++;
        vector<int> res;
        for (int num:nums2)
        {
            if (mp.count(num))
            {
                // 填一个然后
                res.push_back(num);
                // 统计次数-1
                mp[num] --;
                // 当这个mp[num]=0的时候
                // 就不再需要它了 然后就要删掉
                // 防止重复统计
                if (mp[num] == 0) mp.erase(num);
            }            
            
        }
        return res;
    }
    
};

对应我的csdn文章:(2条消息) 两个数组的交集 II(C++)_极客李华的博客-CSDN博客