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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ghscarecrow/article/details/82151527

此题出自牛客网的剑指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;
    }
};

猜你喜欢

转载自blog.csdn.net/ghscarecrow/article/details/82151527