给一个整数数组,找到两个数使得他们的和等于一个给定的数 target。
你需要实现的函数twoSum
需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是 1 到 n,不是以 0
开头。
样例
给出 numbers = [2, 7, 11, 15]
, target = 9
, 返回 [1, 2]
.
solution:
方法一:使用map容器来存储对应的元素和它的下标。
class Solution {
public:
/*
* @param numbers : An array of Integer
* @param target : target = numbers[index1] + numbers[index2]
* @return : [index1+1, index2+1] (index1 < index2)
*/
vector<int> twoSum(vector<int> &nums, int target) {
// write your code here
map <int,int> valueIndex;//对应元素和它的下标
int n = nums.size();
for (int i = 0; i < n; ++i) {
int value = target - nums[i];
auto iter = valueIndex.find(value);//查找差值
if(iter !=valueIndex.end()) { //注意下标是从1开始的
return {iter -> second, i+1};//找到则返回
}
valueIndex.insert({nums[i], i+1});//没有找到,则把元素插入到map中
}
return {};
}
};
建立map<int,int>,用于存储数组中的每个数和它的下标
循环:对每个元素numbers[i],计算其与target的差值value,在map中搜寻value,
如果找到了,则返回迭代器iter,返回的迭代器iter与map的类型相同?包含两位,第一位为数值,第二位为下标,下标从0开始还是1?
若没找到,则将元素number[i]插入到map中
例:i=0,value=9-number[0]=9-2=7 (此时map为空)
在map中搜寻value=7失败,将number[0]插入map中,map=[2,0,0......]
i=1,value=9-number[1]=9-7=2 (此时map=[2,0,0......])
在map中搜寻value=2,找到了
返回
方法二:方法一中使用了多余的空间来存储元素和下标,但是在容器中我们可以直接进行查找。
vector<int> twoSum(vector<int> &nums, int target) {
for (int i = 0; i < nums.size(); ++i) {
int value = target - nums[i];
auto iter = find(nums.begin()+i, nums.end(), value);
if(iter != nums.end()) {
return {i+1, iter-nums.begin()+1};
}
}
return {};
}