给一个整数数组,找到两个数使得他们的和等于一个给定的数 target。
你需要实现的函数twoSum
需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是 0 到 n-1。
注意事项
你可以假设只有一组答案。
样例
给出 numbers = [2, 7, 11, 15]
, target = 9
, 返回 [0, 1]
.
挑战
Either of the following solutions are acceptable:
- O(n) Space, O(nlogn) Time
- O(n) Space, O(n) Time
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> &numbers, int target) {
// write your code here
unordered_map<int, int> nums_hash;
vector<int> res;
for (int i = 0; i < numbers.size(); i++) {
nums_hash[numbers[i]] = i;
}
for (int i = 0; i < numbers.size(); i++) {
if (nums_hash.find(target - numbers[i]) != nums_hash.end()) {
res.push_back(i);
res.push_back(nums_hash[target - numbers[i]]);
return res;
}
}
return res;
}
};