题目来源
给一个数组以及一个targe，求组成这个targe的两个数的索引。
我想的是用哈希表记录下，然后看target-num[i]是不是在哈希表里。代码如下：
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> map;
int n = nums.size();
vector<int> res;
for (int i=0; i<n; i++) {
map[nums[i]] = i;
}
for (int i=0; i<n; i++) {
if (map.count(target-nums[i]) != 0 && i != map[target-nums[i]]) {
res.push_back(i);
res.push_back(map[target-nums[i]]);
return res;
}
}
return res;
}
};
然后还可以进行优化，变成一次循环就可以，代码如下：
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> map;
int n = nums.size();
vector<int> res;
for (int i=0; i<n; i++) {
if (map.count(target-nums[i]) != 0) {
res.push_back(i);
res.push_back(map[target-nums[i]]);
return res;
}
map[nums[i]] = i;
}
return res;
}
};
