题目要求
给定一个整数数组 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]
思路
- 用
unordered_map
建立一个从int到int的映射 - 遍历数组元素,如果
target-nums[i]
存在说明有满足要求的两个整数,返回下标即可 - 需要注意的是,应该首先查询哈希表中是否存在
target-nums[i]
,然后将nums[i]
插入到哈希表中,即可保证不会让nums[i]
和自己匹配。
代码
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> hash;
for(int i=0;i<nums.size();i++)
{
if(hash.count(target-nums[i]))
return {hash[target-nums[i]],i};
hash[nums[i]]=i;
}
return {};
}
};