Leetcode 1 Two Sum

原创于 2021-12-16 21:15:08 发布 · 407 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

Leetcode 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了如何使用哈希表优化LeetCode中的两数之和问题，通过unordered_map实现查询和插入操作，提升查找效率。展示了两种不同的循环起点，确保代码的全面性。

Leetcode画图工具：draw.io

在这里插入图片描述
循环中分为两步，先去字典（哈希表）中进行查询，如果有则返回，否则把当前元素及其索引添加到字典中。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> umap;
        umap[nums[0]] = 0;
        vector<int> res;

        for (int i=1; i<nums.size(); ++i) {
            if (umap.count(target-nums[i])) {
                res.push_back(i);
                res.push_back(umap[target-nums[i]]);
                return res;
            }
            umap[nums[i]] = i;
        }

        res.push_back(-1);
        return res;

    }
};

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> umap;
        vector<int> res;

        for (int i=0; i<nums.size(); ++i) {
            if (umap.count(target-nums[i])) {
                res.push_back(i);
                res.push_back(umap[target-nums[i]]);
                return res;
            }
            umap[nums[i]] = i;
        }

        res.push_back(-1);
        return res;

    }
};