題目來源
給一個數(shù)組以及一個targe秉溉,求組成這個targe的兩個數(shù)的索引力惯。
我想的是用哈希表記錄下,然后看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;
}
};
然后還可以進行優(yōu)化父晶,變成一次循環(huán)就可以,代碼如下:
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;
}
};