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
相关产品推荐
相关产品推荐

