两数之和
问题描述
给一个整数数组,找到两个数使得他们的和等于一个给定的数 target。
你需要实现的函数twoSum需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是 1 到 n,不是以 0 开头。
注意事项
你可以假设只有一组答案。
思路:
- 用一个map来保存原数组中元素的位置。
- 判断目标值与遍历的值的差是否在map中存在,并且判断二者的下标,相同的话,说明元素有重。不同的话,直接返回正确的结果。
代码
vector<int> twoSum(vector<int> &nums, int target) {
// write your code here
int i = 0, j = nums.size() - 1;
vector<int> res;
map<int, int> temp;
for (int i = 0; i < nums.size(); i++) {
temp[nums[i]] = i;
}
for (j = 0; j < nums.size(); j++) {
int t = target - nums[j];
if (temp.count(t) && temp[t] != j) {//避免重复
return { j + 1,temp[t] + 1 };
}
}
return res;
}