今天做leetcode的two sum题,题目要求:给出一int数组和一个目标数字,在数组中找到加和等于目标数字的数的下标。
题目看上去很容易理解,思路也很简单:从数组开始进行遍历与当前数字后面的数相加看看和是否与目标数相等。这样的方法(暴力法)下来的时间复杂度是O^2,提交结果为超时。
优化:利用unordered_map对数组进行存储 map<int ,int> 其中key为数组中的值,val为值在数组中的下标。然后在从数组开始循环,将目标值与当前值做减法,得到的值在map中查找,如果找到,则可以返回下标了。值得注意的是要加另一个判断是两个下标是不是相等,例如数组中有3,目标值是6,这个时候如果3在正确值的前面则会返回两个3当前的位置。
上代码
class Solution {
public:
vector<int> twoSum(vector<int> &numbers, int target) {
vector<int> item;
unordered_map<int, int> umap;
int length = numbers.size();
for(int i = 0; i < length; i++){
umap[numbers[i]] = i;
}
for(int j = 0; j < length; j++){
int opd = (target - numbers[j]);
if(umap.find(opd) != umap.end()){
if(umap[opd] != j){
item.push_back(j+1);
item.push_back(umap[opd]+1);
break;
}
}
}
return item;
}
};