[Luogu P2569] [BZOJ 1855] [SCOI2010]股票交易

洛谷传送门

BZOJ传送门

题目描述

最近 $\text{lxhgww}$ 又迷上了投资股票，通过一段时间的观察和学习，他总结出了股票行情的一些规律。

通过一段时间的观察，$\text{lxhgww}$ 预测到了未来 $T$ 天内某只股票的走势，第 $i$ 天的股票买入价为每股 $A{P}_i$，第 $i$天的股票卖出价为每股 $B{P}_i$（数据保证对于每个 $i$，都有 $A{P}_i\ge B{P}_i$），但是每天不能无限制地交易，于是股票交易所规定第 $i$ 天的一次买入至多只能购买 $A{S}_i$ 股，一次卖出至多只能卖出 $B{S}_i$ 股。

另外，股票交易所还制定了两个规定。为了避免大家疯狂交易，股票交易所规定在两次交易（某一天的买入或者卖出均算是一次交易）之间，至少要间隔 $W$ 天，也就是说如果在第 $i$ 天发生了交易，那么从第 $i+1$ 天到第 $i+W$ 天，均不能发生交易。同时，为了避免垄断，股票交易所还规定在任何时间，一个人的手里的股票数不能超过 $\text{MaxP}$。

在第 $1$ 天之前，$\text{lxhgww}$ 手里有一大笔钱（可以认为钱的数目无限），但是没有任何股票，当然，$T$ 天以后，$\text{lxhgww}$ 想要赚到最多的钱，聪明的程序员们，你们能帮助他吗？

输入输出格式

输入格式：

输入数据第一行包括 $3$ 个整数，分别是 $T$，$\text{MaxP}$，$W$。

接下来 $T$ 行，第 $i $行代表第 $i-1$ 天的股票走势，每行 $4$ 个整数，分别表示 $A{P}_i,B{P}_i,A{S}_i,B{S}_i$。

输出格式：

输出数据为一行，包括 $1$ 个数字，表示 $\text{lxhgww}$ 能赚到的最多的钱数。

输入输出样例

输入样例#1：

5 2 0
2 1 1 1
2 1 1 1
3 2 1 1
4 3 1 1
5 4 1 1

输出样例#1：

3

说明

对于 $30\%$ 的数据，$0\le W<T\le 50,1\le \text{MaxP}\le 50$

对于 $50\%$ 的数据，$0\le W<T\le 2000,1\le \text{MaxP}\le 50$

对于 $100\%$ 的数据，$0\le W<T\le 2000,1\le \text{MaxP}\le 2000$

对于所有的数据，$1\le B{P}_i\le A{P}_i\le 1000,1\le A{S}_i,B{S}_i\le \text{MaxP}$

解题分析

设$dp[i][j]$表示第$i$天手上还有$j$张股票能赚到的最大值。 那么有如下转移：

1. 这一天啥都不干：$dp[i][j]=dp[i-1][j](j\in [0,MaxP])$

2. 这一天在没有票的基础上买了一些票： $dp[i][j]=min(dp[i][j],-AP[i]\ast j)(j\in [0,AS[i]])$

3. 这一天在第$i-w-1$天的基础上卖了一些票：
   $$\begin{gathered} dp[i][j]=max(dp[i][j],max(dp[i-w-1][k]+(k-j)\ast BP[i]))(k\in [j+1,j+BS[i]]) \\ \to dp[i][j]=max(dp[i][j],max(dp[i-w-1][k]+k\ast BP[i])-j\ast BP[i])(k\in [j+1,j+BS[i]]) \end{gathered}$$
   取值范围是个滑动窗口， 单调队列优化。

4. 这一天在第$i-w-1$天的基础上买了一些票： 和上面的式子差不多。

所以两个单调队列就好了。 总复杂度$O(TMaxP)$。

代码如下：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <algorithm>
#define R register
#define IN inline
#define MX 2005
#define W while
#define gc getchar()
#define ll long long
template <class T>
IN void in(T &x)
{
	x = 0; R char c = gc;
	for (; !isdigit(c); c = gc);
	for (;  isdigit(c); c = gc)
	x = (x << 1) + (x << 3) + c - 48;
}
template <class T> IN T max(T a, T b) {return a > b ? a : b;}
template <class T> IN T min(T a, T b) {return a < b ? a : b;}
int dp[MX][MX];
int T, UP, halt, h, t;
int cost[MX], val[MX], lin[MX], lout[MX], que[MX];
int main(void)
{
	in(T), in(UP), in(halt);
	for (R int i = 1; i <= T; ++i)
	in(cost[i]), in(val[i]), in(lin[i]), in(lout[i]);
	std::memset(dp, 128, sizeof(dp));
	for (R int i = 1; i <= T; ++i)
	{
		for (R int j = 0; j <= lin[i]; ++j) dp[i][j] = max(dp[i - 1][j], -cost[i] * j);
		for (R int j = lin[i] + 1; j <= UP; ++j) dp[i][j] = dp[i - 1][j];
		if (i > halt + 1)
		{
			int tar = i - halt - 1;
			que[h = t = 0] = 0;
			for (R int j = 1; j <= UP; ++j)
			{
				W (h < t && que[h] < j - lin[i]) ++h;
				if (h <= t)
				dp[i][j] = max(dp[i][j], dp[tar][que[h]] - (j - que[h]) * cost[i]);
				W (h <= t && dp[tar][que[t]] + que[t] * cost[i] < dp[tar][j] + j * cost[i]) --t;
				que[++t] = j;
			}
			que[h = t = 0] = UP;
			for (R int j = UP - 1; ~j; --j)
			{
				W (h < t && que[h] > j + lout[i]) ++h;
				if (h <= t)
				dp[i][j] = max(dp[i][j], dp[tar][que[h]] + (que[h] - j) * val[i]);
				W (h <= t && dp[tar][que[t]] + que[t] * val[i] < dp[tar][j] + j * val[i]) --t;
				que[++t] = j;
			}
		}
	}
	printf("%d\n", dp[T][0]);
}