美文网首页
1. 两数之和

1. 两数之和

作者: Hoker_ | 来源:发表于2018-12-06 16:03 被阅读0次
    出于练习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,估计在此场景性能要更好。

    相关文章

      网友评论

          本文标题:1. 两数之和

          本文链接:https://www.haomeiwen.com/subject/yvhocqtx.html