题目:
给一个整数数组,找到两个数使得他们的和等于一个给定的数 target。
你需要实现的函数twoSum
需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是 1 到 n,不是以 0
开头。
注意事项
你可以假设只有一组答案。
样例
给出 numbers = [2, 7, 11, 15]
, target = 9
, 返回 [1, 2]
.
解法1:最简单的方法,两重for循环,时间复杂的O(n^2). 提交显示用时10ms.
附代码:
// 解法1 两重for循环
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
vector<int> ans;
for (int i=0; i<nums.size(); ++i) {
for (int j=i+1; j<nums.size(); ++j) {
if (nums[i] +nums[j] == target) {
return {i+1, j+1};
}
}
}
}
};
解法2:用map标记值和下标,一次遍历,如果map里能找到和当前值和为target的值,找到答案。否则插入进map.时间复杂度O(n), 加上map保存的空间复杂度。
// 解法2:O(n)
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> mp;
for (int i=0; i<nums.size(); ++i) {
int val = target - nums[i];
int index = i + 1;
map<int, int>::iterator iter;
iter = mp.find(val);
if (iter != mp.end()) {
return {iter->second, i+1};
}
mp.insert({nums[i], i+1});
}
}
};
解法3:解法2中的map去掉,直接再vector里找。
// 解法3:O(n)
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
for (int i=0; i<nums.size(); ++i) {
int val = target - nums[i];
int index = i+1;
auto iter = find(nums.begin()+i+1, nums.end(), val);
if (iter != nums.end()) {
return {i+1, iter-nums.begin()+1};
}
}
}
};