出于练习C++的考虑,现开始用C++刷题,题目来自于LeetCode
题目描述:
思路比较简单,最开始直接考虑的是两层for循环,显然是不太科学的,后便想到使用map这种数据结构,就是用现在target-nums[i]值,做为key回头往map里找,找到就返回对应的下标。C++的map之前没用过,还浪费了点小时间。
顺便记录下新了解的map用法:
- 使用count,返回的是被查找元素的个数。如果有,返回1;否则,返回0。由于map中不存在相同元素,所以返回值只能是1或0。
- 使用find,返回的是被查找元素的位置,没有则返回map.end()。
代码:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> map;
int i = 0;
for(;i<nums.size();i++){
if(map.count(target-nums[i])!=0){
break;
}else{
map[nums[i]]=i;
}
}
vector<int> ans;
ans.push_back(map[target-nums[i]]);
ans.push_back(i);
return ans;
}
};
整个用例耗时8ms。
用例耗时
看下4ms的代码长啥样:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> m;
vector<int> res;
for(int i=0; i<nums.size(); i++){
if(m.count(target-nums[i])){
return{i,m[target-nums[i]]};
}
m[nums[i]]=i;
}
return {};
}
};
主要差别还是再他用到了unordered_map,估计在此场景性能要更好。
网友评论