Leetcode1. 两数之和(HOT100)

kitesxian

于 2024-11-26 20:54:52 发布

阅读量333

点赞数 1

CC 4.0 BY-SA版权

分类专栏：编程题题解文章标签：算法

本文链接：https://blog.csdn.net/kitesxian/article/details/144067934

编程题题解专栏收录该内容

67 篇文章

订阅专栏

链接

我的代码：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int,int> hash;
        vector<int>res;
        for(int i = 0;i<nums.size();i++){
            hash[nums[i]] = i;
        }
        for(int i = 0;i<nums.size();i++){
            if(hash.count(target-nums[i])&&hash[target-nums[i]]!=i){//注意加hash[target-nums[i]]!=i目的是防止出现6=3+3这种case
                res.push_back(i);
                res.push_back(hash[target-nums[i]]);
                break;
            }
        }
        return res;
    }
};

更好的代码：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int,int> hash;
        for(int i = 0;i<nums.size();i++){
            if(hash.count(target-nums[i])){
                return {hash[target-nums[i]],i};
            }
            hash[nums[i]] = i;
        }
        return {};
    }
};

也就是一边添加，一边判断，因为题目保证了肯定存在两个不重复元素的和是target，所以遍历一遍之后肯定可以找到。