题目:给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和为目标值 target
的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
分析:两数之和的解法有很多种,但在这里只讨论哈希表的做法。因为我们要找到nums中满足和为target的两个数,假设这两个数是a和b,那么他们满足a+b=target,a=target-b;因此我们可以设置一个哈希表存放target-b,然后遍历nums,如果有元素等于target-b,那么就输出[a, target-a]。如果遍历完也找不到,则输出[]。
这里的哈希结构很明显不适合使用纯粹数组,因为我们nums中的元素值很大。所以我们可以使用unordered_set和unordered_map来实现。
一、unordered_set的具体代码实现
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_set<int> hash;
for(int i = 0; i < nums.size(); i++) {
if(hash.find(nums[i]) != hash.end()) {
return {nums[i], target - nums[i]};
}
hash.insert(target - nums[i]);
}
return {};
}
};
但是由于set只能存储数值信息,不能存储索引信息,所以对于题目要求的返回索引信息是无法实现的。因此不能使用unordered_set。
二、unordered_map具体代码实现
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.find(nums[i]) != hash.end()) {
return {i, hash[nums[i]]};
}
hash[target - nums[i]] = i;
}
return {};
}
};
运行结果如下图所示: