【剑指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).