hdu3401 Trade

传送门
好文章,顶

大致思路和“我要长高”没什么区别。 d p [ i ] [ j ] dp[i][j] 表示第 i i 天手上有份 j j 股票。
值得注意的是通过 d p [ i ] [ j ] = m a x ( d p [ i ] [ j ] , d p [ i 1 ] [ j ] ) dp[i][j]=max(dp[i][j],dp[i-1][j]) 来处理只能在 W W 天之后才能交易的限制。于是只用考虑 d p [ i W 1 ] [ j ] dp[i-W-1][j] 的情况了。
买:
d p [ i ] [ j ] = m a x ( d p [ i W 1 ] [ k ] ( j k ) a p [ i ] , d p [ i ] [ j ] ) ( 1 < = k < j ) dp[i][j]=max(dp[i-W-1][k]-(j-k)*ap[i],dp[i][j])(1<=k<j)
卖:
d p [ i ] [ j ] = m a x ( d p [ i W 1 ] [ k ] + ( k j ) b p [ i ] , d p [ i ] [ j ] ) ( j < k < m a x p ) dp[i][j]=max(dp[i-W-1][k]+(k-j)*bp[i],dp[i][j])(j<k<maxp)

对于买,展开后,单调队列维护 m a x ( d p [ i W 1 ] [ k ] + k a p [ i ] ) max(dp[i-W-1][k]+k*ap[i]) ,卖同理。
最后,注意初始化。前面1~(W+1)天只能买进。

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e3+10;
const int oo=1<<30;
int orz,T,maxp,W,ans,head,tail;
int ap[maxn],bp[maxn],as[maxn],bs[maxn],dp[maxn][maxn];
struct Q{
	int val,num;
	Q(){}
	Q(int val,int num):val(val),num(num){}
}q[maxn];
inline int read(){
	int x=0;char ch=getchar();
	while(!isdigit(ch)) ch=getchar();
	while(isdigit(ch)) x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
	return x;
}
int main(){
	orz=read();
	while(orz--){
		T=read(),maxp=read(),W=read(),ans=-oo;
		for(int i=1;i<=T;++i) ap[i]=read(),bp[i]=read(),as[i]=read(),bs[i]=read();
		for(int i=1;i<=T;++i){for(int j=1;j<=maxp;++j) dp[i][j]=-oo;dp[i][0]=0;}
		for(int i=1;i<=W+1;++i) for(int j=0;j<=as[i];++j) dp[i][j]=-j*ap[i];
		for(int i=2;i<=T;++i){
			for(int j=0;j<=maxp;++j) dp[i][j]=max(dp[i-1][j],dp[i][j]);
			if(i<=W+1) continue;
			head=0,tail=0;
			for(int j=0;j<=maxp;++j){
				int f=dp[i-W-1][j]+ap[i]*j;
				while(head<tail&&q[tail-1].val<f) tail--;
				q[tail++]=Q(f,j);
				while(head<tail&&(j-q[head].num>as[i])) head++;
				dp[i][j]=max(dp[i][j],q[head].val-ap[i]*j);
			}
			head=0,tail=0;
			for(int j=maxp;j>=0;--j){
				int f=dp[i-W-1][j]+bp[i]*j;
				while(head<tail&&q[tail-1].val<f) tail--;
				q[tail++]=Q(f,j);
				while(head<tail&&(q[head].num-j>bs[i])) head++;
				dp[i][j]=max(dp[i][j],q[head].val-bp[i]*j);
			}
		}
		for(int i=0;i<=maxp;++i) ans=max(ans,dp[T][i]);
		printf("%d\n",ans);
	}
}

猜你喜欢

转载自blog.csdn.net/g21wcr/article/details/88756834