Lintcode 56. 两数之和

给一个整数数组,找到两个数使得他们的和等于一个给定的数 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 twoSum(vector &numbers, int target) {
        // write your code here
        vector result(2, 0);
        unordered_map 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,此时是有解的,所以按照代码所示方法解决该问题。

你可能感兴趣的:(刷题)