您的位置:首页 > 职场人生

一道亚马逊面试题(股票买卖问题)

2014-03-20 14:21 309 查看
题目:一个股价序列,已知每个时间点的股价,问什么时候买什么时候卖获利最大

这道题O(N)空间复杂度和O(N)时间复杂度都不难想到:假设股价序列由a
表示,一种方法是开一个缓冲数组b[N-1],记录a
中每两个相邻元素的差值,然后问题转化为求b[N-1]的最大子数组和(当然必须知道最大子数组和的O(N)解法)。第二种方法是开一个缓冲数组b
,依次存储以当前元素结尾的左侧数组最小值,然后再次扫描数组,求出a[i]-b[i]的最大值,并进一步得到买入和卖出时间。

上述两种方法都可以不使用缓冲数组,对于第一种方法,在求子数组最大和的过程中直接采用a
的差值,不单独保存起来。对于第二种方法,使用一个变量来保存以当前元素结尾的左侧数组最小值,在向右的扫描过程中不断更新;这样不仅空间复杂度降低,而且扫描只需要一遍。

这里采用第二种方法,形成代码:

#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
 = {20,40, 40, 90, 70};

	cout << MaxProfit(a) << endl;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: