题目描述:
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
输出描述:
对应每个测试案例,输出两个数,小的先输出。
思路分析:
1. 最直接的分析是对于数组中的每个数,都去和它后面的数相加,判断是否和为S。那这样的时间复杂度为O(n^2),超时。
2. 其实思路很简单,但是自己一直没想到。用头尾两个指针,由于数组有序,那么第一个元素就最小,最后一个元素就最大。将当前两个指针所指的元素相加,判断若大于S,则右指针前移,若小于S,则左指针后移。而当相等时,就判断是否为第一对满足情况的数,若是直接保存,否则再进行一次两对满足要求数的乘积比较,保留较小的。这样的时空复杂度都仅为O(n)。
代码:
直接放后一个思路的代码。
class Solution {
public:
vector<int> FindNumbersWithSum(vector<int> array,int sum) {
vector<int> res;
int len = array.size();
if(len<)
return res;
int l,r;
l = ;
r = len-;
while(l<r)
{
if(array[l]+array[r]>sum)
{
r--;
}
else if(array[l]+array[r]<sum)
{
l++;
}
else
{
if(res.size()==)
{
res.push_back(array[l]);
res.push_back(array[r]);
}
else
{
if(res[]*res[]>array[l]*array[r])
{
res[] = array[l];
res[] = array[r];
}
}
l++;
r--;
}
}
return res;
}
};