和为S的两个数字

题目描述

输入一个递增排序的数组和一个数字S，在数组中查找两个数，使得他们的和正好是S，如果有多对数字的和等于S，输出两个数的乘积最小的。

输出描述:

对应每个测试案例，输出两个数，小的先输出。

//Way1:暴力枚举
class Solution {
public:
    vector<int> FindNumbersWithSum(vector<int> array,int sum) {
          vector<int>ret;
        if(array.size()<2) return ret;
        if(array.size()==2) 
        {
            if(array[0]+array[1]==sum)
            {
                     ret.push_back(array[0]);
                      ret.push_back(array[1]);
                      return ret;
            }
        }
          for(int i=0;i<array.size()-1;++i)
              for(int j=i+1;j<array.size();++j)
              {
                  if(array[i]+array[j]==sum) 
                  {
                      ret.push_back(array[i]);
                      ret.push_back(array[j]);
                      return ret;
                  }
              }
        return ret;
    }
};

/*Way2:
数列满足递增，设两个头尾两个指针i和j，
若ai + aj == sum，就是答案（相差越远乘积越小）
若ai + aj > sum，aj肯定不是答案之一（前面已得出 i 前面的数已是不可能），j -= 1
若ai + aj < sum，ai肯定不是答案之一（前面已得出 j 后面的数已是不可能），i += 1
O(n)*/

typedef vector<int> vi;
class Solution {
public:
    vi FindNumbersWithSum(const vi& a,int sum) {
        vi res;
        int n = a.size();
        int i = 0, j = n - 1;
        while(i < j){
            if(a[i] + a[j] == sum){
                res.push_back(a[i]);
                res.push_back(a[j]);
                break;
            }
            while(i < j && a[i] + a[j] > sum) --j;
            while(i < j && a[i] + a[j] < sum) ++i;
        }
        return res;
    }
};