LeetCode-1:两数之和

一、题目描述

在这里插入图片描述

二、解题思路

这道题当然可以暴力,这样时间复杂度为O(n2)O(n^2)。所以我采用哈希表来解决。
遍历一次数组,将其与下标的pair(val,index)pair(val, index)插入到哈希表中。所以我们根据当前的数字去找targetnums[i]target - nums[i]是否在哈希表里即可。但是如果出现targetnums[i]==nums[i]target - nums[i] == nums[i]的情况时,哈希表会根据keykey查找到与当前的nums[i]nums[i]相同的下标indexindex,导致出错,所以只有在targetnums[i]target - nums[i]作为keykey时映射出来的元素的下标与当前元素不等时,才证明找到。

三、解题代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> ump;    //val, index
        vector<int> sln;
        for(int i = 0; i < nums.size(); i++){
            if(ump.find(nums.at(i)) == ump.end())
                ump.insert(make_pair(nums.at(i), i));
            if(ump.find(target - nums.at(i)) != ump.end()){
                if(ump.find(target - nums.at(i))->second != i){
                    sln.push_back(ump.find(target - nums.at(i))->second);
                    sln.push_back(i);
                    return sln;
                }   
            }
        }
        return sln;
    }
};

四、运行结果

在这里插入图片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章