【题解】HDU3401 Trade 单调dp

你是一个股神，可以预测未来n天股票的价格，你最多持有m个股票，且每次交易之后需要间隔w天再进行交易。给定未来n天的每天股票的买入价格pi，卖出价格po，买入限制mi，卖出价值mo，假设初始资金无限，求最多能赚多少钱。

DP
状态表示：$dp[i][j]$表示第i天持有j股票时最多赚多少钱.
状态边界：$dp[i][j]=-inf,dp[0][0]=0.$
状态转移：

1. 不买不卖，$dp[i][j]=dp[i-1][j]$
2. 买入，$dp[i][j]=dp[i-w-1][k1]-(j-k1)\ast pi$，其中$max(0,j-mi)<=k1<=j$
3. 卖出，$dp[i][j]=dp[i-w-1][k2]+(k2-j)\ast po$，其中$j<=k2<=min(m,j+mo)$

单调优化
$dp[i][j]=max(dp[i-w-1][k1]+k1\ast pi)-j\ast pi$
$dp[i][j]=max(dp[i-w-1][k1]+k2\ast po)-j\ast po$

代码

/* LittleFall : Hello! */
#include <bits/stdc++.h>
using namespace std; using ll = long long; inline int read();
const int M = 2048, MOD = 1000000007;

int dp[M][M]; //dp[i][j]表示第i天持有j股票时的最大收益
int A[M], B[M]; //单调队列
int main(void)
{
	#ifdef _LITTLEFALL_
	freopen("in.txt","r",stdin);
    #endif

	int T = read();
	while(T--)
	{
		memset(dp, 0xc0, sizeof(dp)); //负的极小值
		int n=read(), m=read(), w=read(); //天数, 最大钱数, 间隔

		// 初始状态
		dp[0][0] = 0;
		for(int i=1; i<=n; ++i)
		{
			int pi=read(), po=read(), mi=read(), mo=read(); 
			for(int j=0; j<=m; ++j)
				dp[i][j] = dp[i-1][j]; //不买不卖
			if(i<=w+1) 
			{
				for(int j=0; j<=mi; ++j)
					dp[i][j] = max(dp[i-1][j], -j*pi); //只能买
				continue;
			}		
			int l=0, r=0; //单调队列的队头与队尾(左闭右开)	
			for(int j=0; j<=m; ++j) //买入转移到现在
			{
				int val = dp[i-w-1][j] + j*pi;
				while(l<r && B[r-1]<val) r--;
				A[r]=j; B[r++]=val;
				while(l<r && A[l]+mi<j) l++;
				dp[i][j] = max(dp[i][j], B[l]-j*pi);
			}
			l = r = 0;
			for(int j=m; j>=0; --j) //卖出转移到现在
			{
				int val = dp[i-w-1][j] + j*po;
				while(l<r && B[r-1]<val) r--;
				A[r]=j; B[r++]=val;
				while(l<r && A[l]-mo>j) l++;
				dp[i][j] = max(dp[i][j], B[l]-j*po);
			}
		}
		printf("%d\n",dp[n][0]);
	}

    return 0;
}


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;
}