1. 程式人生 > >LintCode 56. 兩數之和

LintCode 56. 兩數之和

com index 找到 owin zh-cn n) sum www. 範圍

題目:

給一個整數數組,找到兩個數使得他們的和等於一個給定的數 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; } };

LintCode 56. 兩數之和