Leo_h

2333

【HDU3401】Trade

题目大意

股票交易共t天,一个人至多持有p股,每次买、卖后的w天内不能进行交易,即第i天交易后第(i+w+1)天才可继续交易
给出每天股票买入和卖出价格,以及每天买入和卖出的股数限制,假设人的资金无限(即买入不受资金不足的限制),最多可赚多少钱?
(t,p,w,ap[i],bp[i],as[i],bs[i]<=2000)

题解

t和p不大,容易想到dp
设dp[i][j]表示第i天持有j股的最大收益
则容易想到dp[i][j]转移为:
(1)不交易 dp[i-1][j]
(2)买入 for k in (j-as[i])~(j-1) dp[i-w-1][k]-ap[i]*(j-k) 中的最大值
(3)卖出 for k in (j+1)~(j+bs[I]) dp[i-w-1][k]+bp[I]*(k-j)中的最大值
以上三种情况取最大值
其中时间无需枚举的原因是:如果第pre天上次交易,转移到第now天时可以先通过转移(1)不变地转移到第(now-w-1)天,再转移到第now天,从而无需枚举pre
至此为止时间效率O(t*p^2) 会TLE
考虑买入,dp[i-w-1][k]-ap[i]*(j-k)=(dp[i-w-1][k]+ap[i]*k)-ap[i]*j
i不变时,粗体部分只与k有关,因此可以枚举k并记录下来其中最大值,在转移到dp[i][j]时直接使用此最大值,这两个循环是并列而不是嵌套,从而降下一维
考虑到as,需要动态删除k<j-as[i]的情况,因此需要单调队列。实现的时候可以把k和j放在同一循环中,因为一定是编号小的状态影响编号大的状态(卖出相反)

关于单调队列的实现,请百度模板题poj2823

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxt 2005
#define maxs 2005
#define lastdeal i-w-1
int T,t,v,w;
int dp[maxt][maxs];
int ap[maxt];
int bp[maxt];
int as[maxt];
int bs[maxt];
struct dddl//单调队列 
{
	int q[maxs];
	int p[maxs];
	int h,t;
	void clear(){h=1,t=0;}
	void eliminate(int minp){while(p[h]<minp&&h<=t) h++;}
	void insert(int pos,int val)
	{
		while(q[t]<=val&&h<=t) t--;
		q[++t]=val;
		p[t]=pos;
	}
	int top(){return q[h];}
}que;
int main()
{
 	scanf("%d",&T);
 	while(T--)
 	{
 		scanf("%d%d%d",&t,&v,&w);
 		for(int i=1;i<=t;i++)
 			scanf("%d%d%d%d",ap+i,bp+i,as+i,bs+i);
 		memset(dp,128,sizeof(dp));
 		for(int i=1;i<=w+1;i++)//注意处理边界 
 			for(int j=0;j<=v;j++) 
 				if(j<=as[i])dp[i][j]=max(dp[i-1][j],-ap[i]*j);
 				else dp[i][j]=dp[i-1][j];
 		for(int i=w+2;i<=t;i++)
 		{
 			for(int j=0;j<=v;j++)
 				dp[i][j]=dp[i-1][j];//昨日不交易 
 			que.clear();//买 
			for(int j=0;j<=v;j++) 	
			{
				que.eliminate(j-as[i]);
				if(j) dp[i][j]=max(dp[i][j],que.top()-ap[i]*j);
				que.insert(j,dp[lastdeal][j]+ap[i]*j);
			}
 			que.clear();//卖
			for(int j=v;j>=0;j--)
			{
				que.eliminate(v-j-bs[i]);
				if(v!=j)dp[i][j]=max(dp[i][j],que.top()-bp[i]*j);
				que.insert(v-j,dp[lastdeal][j]+bp[i]*j);
			}
		}
		printf("%d\n",dp[t][0]);
	}
}


阅读更多
版权声明:本文为博主原创文章,转载请注明源网址blog.csdn.net/leo_h1104 https://blog.csdn.net/Leo_h1104/article/details/51758272
文章标签: OI DP
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

【HDU3401】Trade

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭