HDU 3401 Trade

题意：给出t天内每天股票的买入价格，卖出价格，最多购买数和最多卖出数，求t天后最多能赚多少、

思路：用dp[i][j]表示第i天时手里还剩j支股票的最大获利。

第i天买入时：dp[i][j] = max{dpi-1][k]-(j-k)*api};即dp[i][j] = max{dp[i-1][k]+k*api}-j*api，（0<=k<=j）。

第i天卖出时：dp[i][j] = max{dpi-1][k]-(j-k)*bpi};即dp[i][j] = max{dp[i-1][k]+k*bpi}-j*bpi，（0<=k<=j）。

第i天不买不卖时：可以想成在买入（卖出）时，买了（卖了）0支股票，故不用单独考虑。

以上是最朴素的想法，复杂度为t*maxp*maxp，时间都过不去，所以需要优化一下，注意看上面两个式子的后半部分，对于更新j时，我们需要的是前i-1天的max{dp[i-1][k]+k*api}（max{dp[i-1][k]+k*bpi}）。所以这里可以用一个单调队列来维护，这样复杂度就降到了t*maxp，可以AC了。

 

#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <map>
#include <vector>
#include <cmath>
#include <stack>
#include <queue>
#include <cstdlib>
#include <algorithm>
using namespace std;
typedef __int64 int64;
typedef long long ll;
#define M 100005
#define max_inf 0x7f7f7f7f
#define min_inf 0x80808080
#define mod 1000000007

int q[2005];
int dp[2005][2005];

int main()
{
	int i , j , T;
	scanf("%d",&T);
	while (T--)
	{
		int t , maxp , w;
		scanf("%d%d%d",&t,&maxp,&w);
		int ap , bp , as , bs;
		memset(dp , min_inf , sizeof dp);
		int ans = 0;
		dp[0][0] = 0;
		for (i = 1 ; i <= t ; i++)
		{
			scanf("%d%d%d%d",&ap,&bp,&as,&bs);
			for (j = 0 ; j <= maxp ; j++)dp[i][j] = max(dp[i][j],dp[i-1][j]);//将前i-1天的最优解先放到第i天。

			int temp = i-w-1;
			if (temp < 0)temp = 0;//若i<=w+1，则相当于从第0天开始买卖
			int front , rear;

			//买入
			front = rear = 0;
			for (j = 0 ; j <= maxp ; j++)
			{
				int k = dp[temp][j]+j*ap;
				while (rear > front && dp[temp][q[rear-1]]+q[rear-1]*ap < k)rear--;
				q[rear++] = j;
				while (front < rear && j-q[front] > as)front++;
				dp[i][j] = max(dp[i][j] ,  dp[temp][q[front]]+q[front]*ap-j*ap);
				ans = max(ans , dp[i][j]);
			}

			//卖出
			front = rear = 0;
			for (j = maxp ; j >= 0 ; j--)
			{
				int k = dp[temp][j]+j*bp;
				while (rear > front && dp[temp][q[rear-1]]+q[rear-1]*bp < k)rear--;
				q[rear++] = j;
				while (q[front]-j > bs)front++;
				dp[i][j] = max(dp[i][j] , dp[temp][q[front]]+q[front]*bp-j*bp);
				ans = max(ans , dp[i][j]);
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}