LintCode(M)两数之和

两数之和

描述
笔记
数据
评测
给一个整数数组，找到两个数使得他们的和等于一个给定的数 target。

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

注意事项

你可以假设只有一组答案。

您在真实的面试中是否遇到过这个题？ Yes
样例
给出 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
        vector<int>v;int i=0,j=nums.size()-1,m=0;
       /* if(nums.size()<2) return v;
        multimap<int,int> m;int k=nums.size();
        for(int i=0;i<k;++i)
        m.insert({nums[i],i+1});
        multimap<int,int>::iterator it=m.begin(),it2=m.end(),it3=it2,it4=it,it5=it2;
        if(it->first>target) return v;

        for(;it!=it2;++it){
            k=it->first;
            if(k<=target)
            {
                for(it4=++it;it4!=it5;++it4){
                  if(it4->first==target-k)
                 {int a=(--it)->second,b=it4->second;
             //    a<b?{a=a+b;b=a-b;a=a-b;}:{a=a;b=b;}
           //  if(a<b){a=a+b;b=a-b;a=a-b;}
        //   if(a<b){int temp=b;b=a;a=temp;}
                     v.push_back(a);
                  v.push_back(b);
                  return v;}
                 if(it4->first>target-k) {it5=it4;break;}
                }
            }
            else break;
        }*/
        sort(nums.begin(),nums.end());
       while(i<j){
           m=nums[i]+nums[j];
           if(m==target) 
             {  v.push_back(i+1);v.push_back(j+1);return v;}
            if(m>target) j--;
             if(m<target) i++;

       }
        return v;
    }
};

可是错误,原因在于我把排序后的顺序当答案,题目要的是原顺序,而不是排序过后的顺序

之后,在我改正这个错误后,我又遇见一个错误,当nums[i]==nums[j]的时候，运用以下代码：

for(auto c:NUM){
           p++;
           if(c==nums[i]) v.push_back(p);
              if(c==nums[j]) v.push_back(p);}

测试数据是0，1，2，0
答案应该是1，4
但是输出的确是1，1，4，4

改正后正确代码：

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
        vector<int>v;int i=0,j=nums.size()-1,m=0;
       /* if(nums.size()<2) return v;
        multimap<int,int> m;int k=nums.size();
        for(int i=0;i<k;++i)
        m.insert({nums[i],i+1});
        multimap<int,int>::iterator it=m.begin(),it2=m.end(),it3=it2,it4=it,it5=it2;
        if(it->first>target) return v;

        for(;it!=it2;++it){
            k=it->first;
            if(k<=target)
            {
                for(it4=++it;it4!=it5;++it4){
                  if(it4->first==target-k)
                 {int a=(--it)->second,b=it4->second;
             //    a<b?{a=a+b;b=a-b;a=a-b;}:{a=a;b=b;}
           //  if(a<b){a=a+b;b=a-b;a=a-b;}
        //   if(a<b){int temp=b;b=a;a=temp;}
                     v.push_back(a);
                  v.push_back(b);
                  return v;}
                 if(it4->first>target-k) {it5=it4;break;}
                }
            }
            else break;
        }*/
        vector<int> NUM(nums);int p=0,q=0;
        sort(nums.begin(),nums.end());
       while(i<j){
           m=nums[i]+nums[j];
           if(m==target) 
            // {  v.push_back(i+1);v.push_back(j+1);return v;}
            break;
            if(m>target) j--;
             if(m<target) i++;

       }
       for(auto c:NUM){
           p++;
           if(c==nums[i]) v.push_back(p);
         if(nums[i]!=nums[j]){
      if(c==nums[j]) v.push_back(p);}}

        return v;
    }
};

之后，我看到更简洁的方案：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> m;
        for (int i = 0; i < nums.size(); ++i) {
            if (m.count(target - nums[i])) {
                return {i, m[target - nums[i]]};
            }
            m[nums[i]] = i;
        }
        return {};
    }
};