题目来源力扣:
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
最容易想到的双重遍历:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> set(2);
for(int i = 0; i < nums.size()-1; i++){
for(int j = i + 1; j < nums.size(); j++){
if(nums[i]+nums[j] == target){
set[0] = i;
set[1] = j;
}
}
}
return set;
}
};
没什么好说的。
还可以使用哈希表降低复杂度:O(n)
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int, int> hashm;
vector<int> set(2);
for(int i = 0; i < nums.size(); i++){
int flag = target - nums[i];
if(hashm.count(flag)){
set[0] = i;
set[1] = hashm[flag];
}
hashm[nums[i]] = i;
}
return set;
}
};