luogu P2569 [SCOI2010]股票交易

背景：

清明节假期结束了…

题目传送门：

https://www.luogu.org/problemnew/show/P2569

题意：

假的模拟一个股票交易的过程（作弊）。
可以在某一天购买或卖出股票，单价为$A{P}_i,B{P}_i$；且限制了购买或卖出的数量$A{S}_i,B{S}_i$；且两次交易的时间至少为$w$天。
求最后能赚多少钱。

思路：

显然是$DP$。
设$f_{i,j}$表示第$i$天结束交易后持有$j$股的最大利润。

则考虑$4$种情况：
$[1]:$凭空买入股票（原来没买）：
则$f_{i,j}=-j\ast A{P}_i$（其中$j\le A{S}_i$）。

$[2]:$又买入股票（原来买过）：
假设原来持有$k$股。
因为两次交易的时间至少为$w$天，所以必须从$f_{i-w-1,k}$转移过来。
则$f_{i,j}=\max (f_{i,j},f_{i-w-1,k}-(j-k)\ast A{P}_i)$（其中$k>0,0<j-k\le A{S}_i$得到$k\in [j-A{S}_i,j)$）。

$[3]:$不操作
则$f_{i,j}=\max (f_{i,j},f_{i-1,j})$。

$[4]:$卖出股票（原来必须持有）
假设原来持有$k$股。
因为两次交易的时间至少为$w$天，所以必须从$f_{i-w-1,k}$转移过来。
则$f_{i,j}=\max (f_{i,j},f_{i-w-1,k}+(k-j)\ast B{P}_i)$（其中$k>0,0<k-j\le B{S}_i$得到$k\in (j,j+B{S}_i]$）。

此时时间复杂度为$\Theta (n^3)$。
怎么优化？
发现$[2][4]$情况太慢，考虑优化这两个。

先考虑$[2]$情况。
$f_{i,j}=\max (f_{i,j},f_{i-w-1,k}-(j-k)\ast A{P}_i)$（其中$k\in [j-A{S}_i,j)$）
拆括号，得到：$f_{i,j}=\max (f_{i,j},f_{i-w-1,k}-j\ast A{P}_i+k\ast A{P}_i)$
稍微移一下项，得到：$f_{i,j}=\max (f_{i,j},f_{i-w-1,k}+k\ast A{P}_i-j\ast A{P}_i)$
在枚举$i,j$时，$(-j\ast A{P}_i)$就是定值了，所以我们要使得$(f_{i-w-1,k}+k\ast A{P}_i)$尽可能大，单调队列优化即可。

$[4]$情况同理。

就$OK$了。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
	int n,m,w;
	int f[2010][2010],que[2010];
int main()
{
	int ap,bp,as,bs;
	scanf("%d %d %d",&n,&m,&w);
	memset(f,-63,sizeof(f));
	for(int i=1;i<=n;i++)
	{
		scanf("%d %d %d %d",&ap,&bp,&as,&bs);
		for(int j=0;j<=as;j++)
			f[i][j]=-j*ap;
		for(int j=0;j<=m;j++)
			f[i][j]=max(f[i][j],f[i-1][j]);
		if(i<=w) continue;
		int head=1,tail=0;
		for(int j=0;j<=m;j++)
		{
			while(head<=tail&&que[head]<j-as) head++;
			if(head<=tail) f[i][j]=max(f[i][j],f[i-w-1][que[head]]-(j-que[head])*ap);
			while(head<=tail&&f[i-w-1][que[tail]]+que[tail]*ap<=f[i-w-1][j]+j*ap)
				tail--;
			que[++tail]=j;
		}
		head=1,tail=0;
		for(int j=m;j>=0;j--)
		{
			while(head<=tail&&que[head]>j+bs) head++;
			if(head<=tail) f[i][j]=max(f[i][j],f[i-w-1][que[head]]+(que[head]-j)*bp);
			while(head<=tail&&f[i-w-1][que[tail]]+que[tail]*bp<=f[i-w-1][j]+j*bp)
				tail--;
			que[++tail]=j;
		}
	}
	printf("%d",f[n][0]);
}