2015小米暑期实习笔试题_风口的猪-中国牛市(dp)

浏览数：47 / 时间：2015年06月11日

风口之下，猪都能飞。当今中国股市牛市，真可谓“错过等七年”。给你一个回顾历史的机会，已知一支股票连续n天的价格走势，以长度为n的整数数组表示，数组中第i个元素（prices[i]）代表该股票第i天的股价。假设你一开始没有股票，但有至多两次买入1股而后卖出1股的机会，并且买入前一定要先保证手上没有股票。若两次交易机会都放弃，收益为0。设计算法，计算你能获得的最大收益。
输入数值范围：2<=n<=100,0<=prices[i]<=100
输入例子:

3,8,5,1,7,8

输出例子:

12

代码：

#include <iostream>
#include <cmath>
#include <vector>

using namespace std;

vector<int> prices;

int calculateMax(vector<int> prices)
{
	int i,nmin,nmax,len,ans;
	len=prices.size();
	int *lhs=new int[len];
	int *rhs=new int[len];

	lhs[0]=0;
	nmin=prices[0];
	for(i=1;i<len;i++)
	{
		lhs[i]=max(lhs[i-1],prices[i]-nmin);
		if(prices[i]<nmin)
			nmin=prices[i];
//		cout<<"lhs "<<i<<" "<<lhs[i]<<endl;
	}

	rhs[len-1]=0;
	nmax=prices[len-1];
	for(i=len-2;i>=0;i--)
	{
		rhs[i]=max(rhs[i+1],nmax-prices[i]);
		if(prices[i]>nmax)
			nmax=prices[i];
//		cout<<"rhs "<<i<<" "<<rhs[i]<<endl;
	}

	ans=0;
	for(i=0;i<len;i++)
		if((lhs[i]+rhs[i])>ans)
			ans=lhs[i]+rhs[i];

	return ans;
}

int main()
{
	int m,len;
	while(cin>>m)
		prices.push_back(m);
	cout<<calculateMax(prices)<<endl;
	return 0;
}

下面是去掉注释的测试截图，可以直观一点

技术分享