一、问题描述
给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和为目标值 target
的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
二、问题分析与实现
因为数组中必存在这两个数,所以本质就是看num[i]有没有target-num[i]与之对应。对应关系用哈希表是最合适的。
将数组中的元素挨个取出,看在哈希表中有无与之满足相加为target的元素,有的话就找到了,没有的话就将其放在哈希表中供后续的数组元素使用。
因为unordered_map的find函数是对关键字搜索,故我们需要将数组元素放在key,元素下标放在value。
代码实现:
#include <unordered_map>
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int len = nums.size();
vector<int> result;
unordered_map<int, int> hashmap;
for(int i = 0; i < len; i++)
{
if(hashmap.find(target-nums[i]) != hashmap.end())
{
result.push_back(i);
result.push_back(hashmap.find(target-nums[i])->second);
return result;
}
hashmap[nums[i]] = i;
}
return result;
}
};