You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

LeetCode 349:解法中mp[x]=1;语句的作用是什么?

LeetCode 349. 两个数组的交集:C++解法中mp[x]=1;语句的意义

我用暴力算法解决了LeetCode 349题(给定两个整数数组nums1和nums2,返回由唯一元素组成的交集数组,元素顺序可任意),但看到一段C++解法代码,想搞懂其中mp[x]=1;语句的具体作用,代码如下:

class Solution { 
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        map<int,int> mp;
        vector<int> ans;
        for(int x : nums1)
            mp[x]=1;
        
        for(int x: nums2)
            if(mp[x]==1)
                mp[x]++;
    
        for(auto x: mp)
            if(x.second>1)
                ans.push_back(x.first);
        return ans;
    }   
};

关于mp[x]=1;的具体意义:

  • 这里的mp是map<int, int>类型,键对应数组中的元素值,值用来做存在性和交集标记。
  • 遍历nums1时执行mp[x]=1,核心作用是标记该元素x存在于nums1中,同时利用map的特性:同一个键只会存储一次,所以即使nums1里x出现多次,最终mp[x]都会被设置为1,自动完成了nums1的去重,不用额外处理重复元素。
  • 后续遍历nums2时,判断mp[x]==1就代表x同时存在于nums1中,此时把mp[x]递增(变成2),用来标记这个元素是两个数组的交集成员。
  • 最后遍历map,收集所有值大于1的键,就是去重后的交集结果。

额外提一句:如果追求更高效率,可以把map换成unordered_map,因为后者是哈希表实现,插入和查找的平均时间复杂度是O(1),而map是红黑树实现,时间复杂度是O(logn),题目不要求结果有序,所以两种容器都适用。

内容的提问来源于stack exchange,提问作者Yash Sinha

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.20 01:40:32