『单调队列优化DP』股票交易

$\mathrm{P}\mathrm{r}\mathrm{o}\mathrm{b}\mathrm{l}\mathrm{e}\mathrm{m}$

$\mathrm{S}\mathrm{o}\mathrm{l}\mathrm{u}\mathrm{t}\mathrm{i}\mathrm{o}\mathrm{n}$

设f[i][j]表示前i天拥有j个股票的拥有的最多的前。

显然有如下几种情况：

• 如果当前不进出股票
  $$f[i][j]=\max (f[i][j],f[i-1][j])$$
• 所有股票都是今天买的
  $$f[i][j]=\max (f[i][j],j\ast A{p}_i)$$
• 买股票
  $$f[i][j]=\underset{j-A{s}_i\le k<j}{\max }(f[i][j],f[i-w-1][k]-(j-k)\ast A{p}_i)$$
• 卖股票
  $$f[i][j]=\underset{j<k\le j+B{s}_i}{\max }(f[i][j],f[i-w-1][k]+(k-j)\ast B{p}_i)$$

这显然是很简单了，我们再来考虑一下如何单调队列。

首先对于买股票的操作，单调队列一开始肯定要是空的，因为没有决策会小于0的。

对于卖股票的操作，由于需要逆序的操作，而且一开始的m一定在$j+B{s}_i$的范围内，所有要有一个$m$。

$\mathrm{C}\mathrm{o}\mathrm{d}\mathrm{e}$

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;
const int N = 2100;

int n, m, w;
int Ap[N], Bp[N], As[N], Bs[N], f[N][N], q[N];

int read(void)
{
	int s = 0, w = 0; char c = getchar();
	while (c < '0' or c > '9') w |= c == '-', c = getchar();
	while (c >= '0' and c <= '9') s = s*10+c-48, c = getchar();
	return w ? -s : s;
}

int main(void)
{
	freopen("test.in","r",stdin);
	freopen("test.out","w",stdout);
	n = read(), m = read(), w = read();
	for (int i=1;i<=n;++i)
		Ap[i] = read(), Bp[i] = read(), As[i] = read(), Bs[i] = read();
		
	memset(f,-30,sizeof f); 
	f[0][0] = 0;
	for (int i=1;i<=n;++i)
	{
		for (int j=0;j<=As[i];++j) f[i][j] = -j * Ap[i]; 
		for (int j=0;j<=m;++j) f[i][j] = max(f[i][j],f[i-1][j]);
		if (i - w - 1 < 0) continue;
		#define calc1(e) (f[i-w-1][e]+e*Ap[i])
		#define calc2(e) (f[i-w-1][e]+e*Bp[i])
		int h = 1, t = 0;
		for (int j=0;j<=m;++j)
		{
			while (h <= t and q[h] < j-As[i]) h ++;
			if (h <= t) f[i][j] = max(f[i][j],calc1(q[h]) - j * Ap[i]);
			while (h <= t and calc1(j) >= calc1(q[t])) t --;
			q[++t] = j;
		}
		h = 1, t = 1;q[1] = m;
		for (int j=m;j>=0;--j)
		{
			while (h <= t and q[h] > j+Bs[i]) h ++;
			if (h <= t) f[i][j] = max(f[i][j],calc2(q[h]) - j * Bp[i]);
			while (h <= t and calc2(j) >= calc2(q[t])) t --;
			q[++t] = j;
		}
	}
	
	printf("%d\n", f[n][0]);
	return 0;
}