Q:输入一个递增排序的数组和一个数字S,在数组中查找两个数,是的他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
思路:
1,因为数组满足递增,从首尾开始遍历 i,j,如果ai+aj=S,就是答案 (相差越大,乘积越小,所以第一个遇到就是乘积最小的)
2,如果 ai+aj>S, aj肯定不是答案之一,j-- (aj后面的不可能,和只会更大)
3,如果 ai+aj<S, ai肯定不是答案之一,i++ (ai前面的不可能,和只会更小)
复杂度O(n)
题目关键在于证明:和为定值的两个数,相差越远乘积越小。
首先是图形解释(只适用于 S>0)
下面做数学上的证明:
所取的两个数分别为 x 和 S-x
那么乘积可表示为:
两个数的差 , 用y表示x得到
进一步乘积可以表示为:
显然,乘积 f 是随 y单调递减的,即可证明再实数范围内,和为定值的两个数 差越大,他们的乘积则越小。
简单的一个demo如下:
#include "stdafx.h"
#include<vector>
#include<iostream>
using namespace std;
vector<int> findMinmul4Sum(vector<int> &arr, int sum)
{
vector<int> result;
int n = arr.size();
int i = 0, j = n - 1;
while (i < j)
{
if (arr[i] + arr[j] == sum) { result.push_back(arr[i]);result.push_back(arr[j]);break; }
if (arr[i] + arr[j] < sum) i++;
if (arr[i] + arr[j] > sum) j--;
}
return result;
}
int main()
{
vector<int> a = {1,2,3,4,5,6,7,8,9};
vector<int> b= findMinmul4Sum(a, 8);
cout << b[0] <<"\t"<<b[1] << endl;
system("pause");
return 0;
}
循环部分可以进一步优化为
while(i<j)
{
if(arr[i]+arr[j]==sum) ...
while(arr[i]+arr[j]<sum && i<j) i++; //i<j必须
while(arr[i]+arr[j]>sum && i<j) j--;
}
速度可以进一步加快