LintCode 56. 两数之和

Posted 2020-10-22 zslhg903

题目：

给一个整数数组，找到两个数使得他们的和等于一个给定的数 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

 

解：要找A+B=target，可以在知道A的情况下，直接找target-A是否存在。

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> res;
        map<int,int> m;
        int sz=numbers.size();
        for(int i=0;i<sz;i++)
        {
            if(m.find(target-numbers[i])!=m.end())
            {
                res.push_back(m[target-numbers[i]]);
                res.push_back(i);
                return res;
            }
            m[numbers[i]]=i;
        }
        return res;
    }
};