P2569 [SCOI2010] 股票交易题解

股票交易题解

大致思路：

这道题一眼动态规划，我们来设一个状态，设 $d{p}_{i,j}$ 为前 $i$ 天拥有股票数量 $j$ 的最大收益，而状态转移方程需要分类讨论。

• 一直单纯买，那么转移方程很明显，一开始我们会将所有数据初始为极小的数，为 $d{p}_{i,j}=j\times a{p}_i$。

• 不买也不卖，那么状态转移方程由上面方法转移而来，非常明显，与前一天中找收益大的，转移方程为 $d{p}_{i,j}=\max (d{p}_{i,j},d{p}_{i-1,j})$。

• 在上面两种方法的情况下买股票，这个转移方程稍微复杂一些，由于每次买卖之间需要相隔 $w$ 天，也就是上一次进行是在第 $i-w-1$ 天，我们假设第 $i-w-1$ 天拥有 $k$ 张股票，而 $k$ 一定比 $j$ 小，因为股票买的要尽可能多，所以，但又不能太多，限制最多 $as$ 张，所以底线是 $j-as$，而本次共买了 $j-k$ 张股票，要用去 $(j-k)\times a{p}_i$ 元，那么转移方程就是 $d{p}_{i,j}=\max (d{p}_{i-w-1,k}-(j-k)\times a{p}_i,d{p}_{i,j})$。

• 在最上面两种情况下卖股票，道理与上一样，就不细讲了，得出转移方程 $d{p}_{i,j}=\max (d{p}_{i-w-1,k}+(j-k)\times b{p}_i,d{p}_{i,j})$。

而以上转移方程，需要在 $O(n^3)$ 的时间复杂度下完成，一看数据 $2000^3$ 肯定过不了。

那么我们需要优化，在下面两种情况中优化，因为那些方程符合单调性优化，以第三种情况为准，用分配律 $d{p}_{i,j}=\max (d{p}_{i-w-1,k}+k\times a{p}_i,d{p}_{i,j})-j\times a{p}_i$。第四种情况也是如此，再用单调队列优化，现在时间复杂度降为了 $O(n)$，可以通过。

代码实现：

#include <bits/stdc++.h>
#define int long long
#define fo(i,x,y) for(int i = x;i <= y;i ++)
#define fr(i,x,y) for(int i = x;i >= y;i --)
#define in(p) cin >> p
#define out(p) cout << p
#define k(n) puts("qwq")
using namespace std;
const int N = 2005;
const int MOD = 123456789;
int mx = 0;
int mn = 0x3f3f3f;
int n, maxp, w, dp[N][N], ap, bp, as, bs, res;
deque <int> q;
inline int read(){
	int r = 0,w = 1;
	char c = getchar();
	while (c < '0' || c > '9'){
		if (c == '-'){
			w = -1;
		}
		c = getchar();
	}
	while (c >= '0' && c <= '9'){
		r = (r << 3) + (r << 1) + (c ^ 48);
		c = getchar();
	}
	return r * w;
}
void write(int x){
	if(x < 0){
		putchar('-');
		x = -x;
	}
	if(x > 9) write(x / 10);
	putchar(x % 10 + '0');
	return;
}
bool cmp(int l, int r)
{
	return l > r;
}

signed main()
{
	n = read();
	maxp = read();
	w = read();
	memset(dp, -0x3f, sizeof dp);
	q.clear();
	for(int i = 1;i <= n;i ++)
	{
		int l = 1, r = 0; 
		ap = read();
		bp = read();
		as = read();
		bs = read();
		for(int j = 0;j <= as; j ++)//方案1 
		{
			dp[i][j] = (-ap) * j;
		}
		for(int j = 0;j <= maxp;j ++)//方案2 
		{
			dp[i][j] = max(dp[i][j], dp[i - 1][j]);
		}
		if(i <= w) continue;//防越界
		
		for(int j = 0;j <= maxp;j ++) //方案3 
		{
			while(q.size() && q.front() < j - as)	q.pop_front();
			while(q.size() && dp[i - w - 1][q.back()] + q.back() * ap <= dp[i - w - 1][j] + j * ap) q.pop_back();
			q.push_back(j);
			if(q.size()) dp[i][j] = max(dp[i][j], dp[i - w - 1][q.front()] + ap * q.front() - j * ap);
		}
		q.clear();
		l = 1, r = 0;
		for(int j = maxp;j >= 0;j --)//方案4 
		{
			while(q.size() && q.front() > j + bs)	q.pop_front();
			while(q.size() && dp[i - w - 1][q.back()] + q.back() * bp <= dp[i - w - 1][j] + j * bp) q.pop_back();
			q.push_back(j);
			if(q.size()) dp[i][j] = max(dp[i][j], dp[i - w - 1][q.front()] + bp * q.front() - j * bp);
		}
		q.clear();
	}
	for(int i = 0;i <= maxp;i ++)
	{
		res = max(res, dp[n][i]);
		//cout << dp[n][i] << "\n";
	}
	write(res);
	return 0;
	k(1);
}