题目描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
代码样例:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
//输入代码
}
};
思路:
最简单的就是双重for循环,但大概率会时间超限,
可以使用哈希表,哈希表的key是数组元素,value是数组下标1,来进行查询,就只需要一次for循环。
具体操作:
遍历开始时,给一个空的哈希表,也就是unordered_map<int,int>
,此时开始遍历nums[]
数组,当遍历到第i
个元素时,查询哈希表中是否存在一个值target - nums[i]
。如果没有,则将当前的nums[i]
和i
存入哈希表;如果有,将当前的i
和map[target - nums[i]]
存入一个pair中。最后以数组的形式返回pair的两个值。这里要注意了,存入的两个下标哪个在前,哪个在后,我分别试了一下,并没有影响,提交都是通过。
代码:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> dict;
pair<int,int> answer;
for(int i = 0;i < nums.size();i++)
{
if(dict.count(target - nums[i]))
{
answer.first = dict[target - nums[i]];
answer.second = i;
break;
}
else
{
dict[nums[i]] = i;
}
}
return {answer.first,answer.second};
}
};