【知识迁移能力】和为S的两个数字

此题出自牛客网的剑指offer专题

题目描述

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

输出描述:

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

解题思路

由于数列满足递增，可以设两个头尾两个指针i和j，

若ai + aj == sum，就是答案（由于相差越远乘积越小，所以此时可以直接跳出循环不用再找了）

若ai + aj > sum，aj肯定不是答案之一（前面已得出 i 前面的数已是不可能），j--

若ai + aj < sum，ai肯定不是答案之一（前面已得出 j 后面的数已是不可能），i++

时间复杂度为O(n)

代码实现

Java版本

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> FindNumbersWithSum(int [] array,int sum) {
        ArrayList<Integer> result = new ArrayList<>();
        int len = array.length;
        if(len<2 || array==null){
            return result;
        }
        int i=0,j=len-1;
        //左右夹逼
        while(i<j){
            if(array[i]+array[j]==sum){
                result.add(array[i]);
                result.add(array[j]);
                break;
            }
            while(i<j && array[i]+array[j]>sum) j--;
            while(i<j && array[i]+array[j]<sum) i++;
            
        }
        return result;
    }
}

C++版本

class Solution {
public:
    vector<int> FindNumbersWithSum(vector<int> array,int sum) {
        vector<int> result;
        if(array.size()<2)
        {
            return result;
        }
        int len = array.size();
        int i=0,j=len-1;
        while(i<j)
        {
            if(array[i]+array[j]==sum)
            {
                result.push_back(array[i]);
                result.push_back(array[j]);
                break;
            }
            while(array[i]+array[j]>sum) j--;
            while(array[i]+array[j]<sum) i++;
            
        }
        return result;
    }
};