给一个整数数组,找到两个数使得他们的和等于一个给定的数target。
你需要实现的函数twoSum需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是1到n,不是以0开头。
样例
numbers=[2, 7, 11, 15], target=9
return [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}; //找到则返回
}
//没有找到,则把元素插入到map中
valueIndex.insert({nums[i], i+1});
}
return {};
}
};
方法二:方法一中使用了多余的空间来存储元素和下标,但是在容器中我们可以直接进行查找。
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 {};
}