hdu 3401 Trade （单调队列优化）

题目链接：trade

单调队列优化dp：解决形如dp[i]=max/min (f[k])+g[i] (满足k<i且g[i]是与k无关的量)，优化的是f[k]。

对于这道题，设dp[i][j]表示在第i天手中有股票数j的最大收益

动态转移方程为：dp[i][j]=max(dp[i-w-1][k]-(j-k)*ap[i]); //买入
                                dp[i][j]=max(dp[i-w-1][k]+(k-j)*bp[i]);//卖出
dp[i][j]=max(dp[i-1][j]);                        //不做交易

先讨论买入：化简为dp[i][j]=max(dp[i-w-1][k]+k*ap[i])-j*ap[i]，即可用单调队列优化。

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define inf (1e7)
int ap[2005],bp[2005],as[2005],bs[2005],dp[2005][2005];
int head,tail;
struct node{int pos,f;}q[2005];

inline int read(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}

int main(){
	int T=read();
	while(T--){
		int N=read(),maxp=read(),w=read();
		for(int i=1;i<=N;i++)
			ap[i]=read(),bp[i]=read(),as[i]=read(),bs[i]=read();
		for(int i=0;i<=N;i++)//初始化
			for(int j=0;j<=maxp;j++)
				dp[i][j]=-inf;
		for(int i=1;i<=w+1;i++)//只买入
			for(int j=0;j<=(min(maxp,as[i]));j++)
				dp[i][j]=-ap[i]*j;
		dp[0][0]=0;
		for(int i=1;i<=N;i++){
			for(int j=0;j<=maxp;j++)
				dp[i][j]=max(dp[i][j],dp[i-1][j]);
			if(i<=w+1)continue;//只能买入的情况
			int pre=i-w-1;//上一次进行交易的天数
			head=tail=0;
			for(int j=0;j<=maxp;j++){
				int temp=dp[pre][j]+j*ap[i];
				while(head<tail&&q[tail-1].f<temp)tail--;
				q[tail].f=temp; q[tail++].pos=j;
				while(head<tail&&q[head].pos+as[i]<j)head++;
				dp[i][j]=max(dp[i][j],q[head].f-j*ap[i]);
			}
			head=tail=0;
			for(int j=maxp;j>=0;j--){
				int temp=dp[pre][j]+j*bp[i];
				while(head<tail&&q[tail-1].f<temp)tail--;
				q[tail].f=temp; q[tail++].pos=j;
				while(head<tail&&q[head].pos-bs[i]>j)head++;
				dp[i][j]=max(dp[i][j],q[head].f-j*bp[i]);
			}
		}
		int ans=0;
		for(int i=0;i<=maxp;i++)
			ans=max(ans,dp[N][i]);
		printf("%d\n",ans);
	}
	return 0;
}