1. 两数之和

出于练习C++的考虑,现开始用C++刷题,题目来自于LeetCode

题目描述:

1. 两数之和

思路比较简单,最开始直接考虑的是两层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,估计在此场景性能要更好。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容