题目:一个股价序列,已知每个时间点的股价,问什么时候买什么时候卖获利最大
这道题O(N)空间复杂度和O(N)时间复杂度都不难想到:假设股价序列由a[N]表示,一种方法是开一个缓冲数组b[N-1],记录a[N]中每两个相邻元素的差值,然后问题转化为求b[N-1]的最大子数组和(当然必须知道最大子数组和的O(N)解法)。第二种方法是开一个缓冲数组b[N],依次存储以当前元素结尾的左侧数组最小值,然后再次扫描数组,求出a[i]-b[i]的最大值,并进一步得到买入和卖出时间。
上述两种方法都可以不使用缓冲数组,对于第一种方法,在求子数组最大和的过程中直接采用a[N]的差值,不单独保存起来。对于第二种方法,使用一个变量来保存以当前元素结尾的左侧数组最小值,在向右的扫描过程中不断更新;这样不仅空间复杂度降低,而且扫描只需要一遍。
这里采用第二种方法,形成代码:
#include <iostream>
using namespace std;
#define N 5
int MaxProfit(int a[])
{
int profit = 0;
int leftmin = a[0];
int buyday = 0;
int sellday = 0;
int tmp = 0;
for (int i=0; i<N; i++)
{
if (a[i] < leftmin)
{
leftmin = a[i];
tmp = i;
}
if (a[i]-leftmin > profit)
{
profit = a[i]-leftmin;
buyday = tmp;
sellday = i;
}
}
cout << "buyday is: " << buyday+1 << endl;
cout << "sellday is: " << sellday+1 << endl;
return profit;
}
void main()
{
int a[N] = {20,40, 40, 90, 70};
cout << MaxProfit(a) << endl;
}