题目
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target){
unordered_map<int, int> mapping;//声明哈希表
vector<int> result;//定义结果向量
for(int i=0; i<nums.size(); i++){
mapping[nums[i]] = i;//初始化哈希表,每个元素对应的是它的下标
}
for(int i=0; i<nums.size(); i++){//遍历数组,寻找每个元素要加的数,使二者之和等于target
const int gap = target - nums[i];//要找的数gap
if(mapping.find(gap)!=mapping.end() && mapping[gap]>i){//找到了gap且这一对数不重复
result.push_back(i);
result.push_back(mapping[gap]);
}
}
return result;
}
};
-
思路:哈希
- 用一个unordered_map存储数组中每个数及对应的下标
- 遍历数组,对每一个数,计算它需要什么数才能凑成target,将这个要寻找的数记为gap。在哈希表中查找是否有键位gap的,只有在条件1成立的情况下才会判断条件二,即确定哈希表中有gap的情况下判断gap的下标要大于i(避免重复)。
- 如果满足以上条件,则表明找到了一组求和数,将二者添加到result向量中。
-
时间复杂度:O(n)
-
空间复杂度:O(n)
-
优化:一次遍历+去重
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> hash_table;//存储{nums[i],i}
//一次遍历,如果当前target-nums[i]没有找到,则将当前数加入哈希表。如果存在当前数和另一个数的和为target,在哈希表中查找当前数时会成功找到
//不用去重,因为一组数中只有先遍历到的数会进入哈希表
for(int i=0;i<nums.size();i++)
{
auto it=hash_table.find(target-nums[i]);//查找当前所需差值
if(it!=hash_table.end())//找到了则返回
{
return {i,it->second};
}
hash_table[nums[i]]=i;//没找到则将当前数加入哈希表
}
return {};
}
};