LintCode_056_两数之和

问题描述

给一个整数数组，找到两个数使得他们的和等于一个给定的数 target。

你需要实现的函数twoSum需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是 1 到 n，不是以 0 开头。

样例

给出 numbers = [2, 7, 11, 15], target = 9, 返回 [1, 2].

求解

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
        std::unordered_map<int, int> hash;
        vector<int> res;
        for(int i = 0; i < nums.size(); ++i) {
            int val = target - nums[i];
            if(hash.find(val) != hash.end()){
                res.push_back(hash[val] + 1);
                res.push_back(i + 1);
                break;
            }
            hash[nums[i]] = i;
        }
        return res;
    }
};