两个数组的交集 II

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

例如:
给定 nums1 = [1, 2, 2, 1]nums2 = [2, 2], 返回 [2, 2].

注意:

  •    输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
  •    我们可以不考虑输出结果的顺序。

思路:对于这种题型,我的第一个思路就是遍历nums2,看nums1中是否有元素与其相等,若相等,则投入交集数组。
但仔细想想就会发现行不通,因为数组中会可能会有重复元素。
既然会有重复元素,我们不妨延伸该思路,你重复,那我就为你保存你出现的次数。怎么保存?很简单,声明一个数组或者STL的map关联容器来保存nums1中每个值出现的次数。然后再用nums2一一比较,若相等,则与该元素关联的次数减一,减为0,就删除该元素。

代码如下:
class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        if (nums1.empty()) 
            return {};
        if (nums2.empty())
            return {};
        vector<int> result;
        map<int,int> nums;
        for (auto i = nums1.cbegin(); i < nums1.cend(); ++i)
        {
            ++nums[*i];
        }
        for (int i = 0; i < nums2.size(); ++i)
        {
            if (nums[nums2[i]])
            {
                result.push_back(nums2[i]);
                if (--nums[nums2[i]] == 0)
                    nums.erase(nums2[i]);
            }
        }
        return result;
    }
};


评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值