leetcode_1-两数之和

题目

描述

给定一个整数数组 nums 和一个整数目标值 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 <= 103
• -109 <= nums[i] <= 109
• -109 <= target <= 109
• 只会存在一个有效答案

代码

#include <unordered_map>
#include <vector>

using namespace std;

class Solution {
public:
    vector<int> twoSum(vector<int> &nums, int target) {
        vector<int> res_vec;
        unordered_map<int, int> value_index_map;
        for (int i = 0; i < nums.size(); ++i) {
            auto iter = value_index_map.find(target - nums[i]);
            if (iter != value_index_map.end()) {
                res_vec.emplace_back(iter->second);
                res_vec.emplace_back(i);
                return res_vec;
            }
            value_index_map.emplace(nums[i], i);
        }
        return res_vec;
    }
};