【剑指Offer-知识迁移能力】面试题57:合为s的两个数字

题目描述

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

思路

我们可以选择数组第一个数字a[left]和数组最后一个数字a[right]作为初始的两个数,这样两个数字的和t=a[left]+a[right]。如果t==s,那么返回这两个数即可。如果t<s,说明我们需要增大t,因为数组是递增的,所以left++就可以增大t(right在数组尾,无法继续增大)。如果t>s,我们需要减小t,也就是right--。重复这一过程,直到left>=right。代码如下:

class Solution {
public:
    vector<int> FindNumbersWithSum(vector<int> array,int sum) {
        vector<int> ans;
        if(array.empty())
            return ans;
        
        int left = 0;
        int right = array.size()-1;
        while(left<right){
            int t = array[left] + array[right];
            if(t==sum){
                ans.push_back(array[left]);
                ans.push_back(array[right]);
                return ans;
            }
            else if(t<sum)
                left++;
            else right--;
        }
        return ans;
    }
};

该算法的时间复杂度为O(n).

猜你喜欢

转载自www.cnblogs.com/flix/p/12525341.html
今日推荐