给一个整数数组,找到两个数使得他们的和等于一个给定的数 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
vector<int> result(2, 0);
unordered_map<int, int> hashMap;
for(int i = 0; i < numbers.size(); hashMap[numbers[i]] = i, i++){ //注意此处,首先注意hashMap[](不是小括号)。并且第一次将i++放在了hashMap[numbers[i] = i]后面,出现答案错误。
if(hashMap.find(target - numbers[i]) == hashMap.end()) continue;
result[0] = hashMap[target - numbers[i]];
result[1] = i;
}
return result;
}
};
写一下上面的想法,之所以先定义空unordered_map然后将遍历过的元素添加进哈希中而不是直接初始化hashMap时就赋值,是因为哈希无法存储相同元素,因为相同的元素会有相同的哈希值。所以当数组为{2, 5, 6},待求值target = 4时无解,但当数组{2, 2, 5, 6},待求值target = 4,此时是有解的,所以按照代码所示方法解决该问题。