给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
#include <iostream>
#include <unordered_map>
#include <vector>
using namespace std;
vector<int> twoSum(vector<int>& nums, int target)
{
unordered_map<int, int> map;
for (int i=0; i<nums.size(); i++)
{
auto iter = map.find(target - nums[i]);
if (iter != map.end())
{
return {iter->second, i};
}
map.emplace(nums[i], i);
}
return {};
}
int main()
{
int target = 9;
vector<int> nums = {2, 3, 11, 7, 16, 13};
vector<int> output = twoSum(nums, 9);
for (int i=0; i<output.size(); i++)
{
cout << output[i] << endl;
}
system("pause");
}