代码随想录_哈希表_leetcode1 两数之和

最新推荐文章于 2024-09-12 19:02:20 发布

thh_1213

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量45

点赞数

分类专栏：代码随想录 leetcode 文章标签：算法数据结构 leetcode

本文链接：https://blog.csdn.net/thh_1213/article/details/129763599

版权

leetcode 同时被 2 个专栏收录

60 篇文章 0 订阅

订阅专栏

代码随想录

59 篇文章 0 订阅

订阅专栏

给定一个整数数组和目标值，找到数组中两个数，使它们相加等于目标值。返回它们的数组下标，要求每个输入有唯一答案，且不能重复使用数组中的元素。提供了一个使用哈希表降低时间复杂度至O(n)的解决方案。

摘要由CSDN通过智能技术生成

题目

1. 两数之和

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target 的那两个整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

提示：

2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案

进阶：你可以想出一个时间复杂度小于 O(n2) 的算法吗？

代码

#include<iostream>
#include<unordered_map>
#include<vector>
using namespace std;
 

// unordered_map   key代表元素 value代表元素对应下标

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> retMap;
        vector<int> result(2, 0);
        for (int i = 0; i < nums.size(); ++i)
        {
            auto it = retMap.find(target - nums[i]);
            if (it != retMap.end())
            {
                result[0] = i;
                result[1] = it->second;
                return result;
            }
            retMap.insert(pair<int, int>(nums[i], i));
        }
        return result;
    }
};