题目描述
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
输出描述:
对应每个测试案例,输出两个数,小的先输出。
题目所给数组为递增排序,所以设置两个指针,分别从头和尾向中间逼近,满足和为sum的两个数之间的距离越远,则其乘积越小,复杂度O(n)。
代码如下:
vector<int> FindNumbersWithSum(vector<int> array,int sum) {
vector<int> res;
auto len=array.size();
if(len-1<2)
return res;
int low=0;
int high=len-1;
int S=0;
while(low<high)
{
S=array[low]+array[high];
if(S>sum)
high--;
else if(S<sum)
low++;
else
{ //首次得到的和为sum的两个数距离一定是最远的
res.emplace_back(array[low]);
res.emplace_back(array[high]);
break;
}
}
return res;
}
另一种方法,复杂度较高,
vector<int> FindNumbersWithSum(vector<int> array,int sum) {
vector<int> res;
if(array.empty())
return res;
auto len=array.size();
int mul=INT_MAX;
for(int i=0;i<len;i++)
{
for(int j=i+1;j<len;j++)
{
if (array[j]>sum)
break;
if(array[i]+array[j]==sum)
{
if(mul>array[i]*array[j])
{
mul=array[i]*array[j];
res.erase(res.begin(),res.end());
res.emplace_back(array[i]);
res.emplace_back(array[j]);
}
}
}
}
return res;
}