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 {};
    }
};