LintCode 56 两数之和

题目:twoSum

---

要求:

给一个整数数组，找到两个数使得他们的和等于一个给定的数 target。
你需要实现的函数twoSum需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是 1 到 n，不是以 0 开头。
注意事项
你可以假设只有一组答案。

样例:

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

算法要求:

Either of the following solutions are acceptable:
O(n) Space, O(nlogn) Time
O(n) Space, O(n) Time

解题思路:

第一次遍历用map根据数值来记录下标。
第二次遍历接着判断对应数是否存在，存在则跳出循环，并输出。

算法如下:

    vector<int> twoSum(vector<int> &nums, int target) {
        // write your code here
        map<int, int> m;
        vector<int> vec;
        int size = nums.size();
        for (int i = 0; i < size; i++) {
            m[nums[i]] = i + 1;
        }
        for (int j = 0; j < size; j++) {
            if (m[target - nums[j]] != 0) {
                vec.push_back(j + 1);
                vec.push_back(m[target - nums[j]]);
                break;
            }
        }
        return vec;
    }