1. 程式人生 > >股票交易[單調佇列優化DP]

股票交易[單調佇列優化DP]

傳送門

我們用f[i][j] 表示交易到第i天,手上有j個股票的最大收益

對於買入

f[i][j] = min(f[i][j],f[i-w-1][k]-(j-k)*a[i])

變一下形

f[i][j]=min(f[i-w-1][k]+k*a[i])-j*a[i]

單調佇列維護min中的一坨就可以了


#include<bits/stdc++.h>
#define N 5005
using namespace std;
int f[N][N];
int T,P,w;
int a[N],b[N] , c[N],d[N];
int q[N];
int main(){
	//freopen("1.in","r",stdin);
	scanf("%d%d%d",&T,&P,&w);
	memset(f,-63,sizeof(f)); f[0][0]=0;
	for(int i=1;i<=T;i++){
		scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
	}
	for(int i=1;i<=T;i++){
		for(int j=0;j<=c[i];j++) f[i][j] = -j*a[i];
		for(int j=0;j<=P;j++) f[i][j] = max(f[i][j] , f[i-1][j]);
		if(i<=w) continue;
		int x=i-w-1; int l=1,r=0;
		for(int j=0;j<=P;j++){
			while(l<=r && j-q[l] > c[i]) l++;
			while(l<=r && f[x][j] + j*a[i] >= f[x][q[r]] + q[r]*a[i]) r--;
		    q[++r] = j;
			if(l<=r) f[i][j] = max(f[i][j] , f[x][q[l]] - (j-q[l])*a[i]); 
		} 
		l=1,r=0;
		for(int j=P;j>=0;j--){
			while(l<=r && q[l]-j > d[i]) l++;
			while(l<=r && f[x][j] + j*b[i] >= f[x][q[r]] + q[r]*b[i]) r--;
			q[++r] = j;
			if(l<=r) f[i][j] = max(f[i][j] , f[x][q[l]] - (j-q[l])*b[i]);
		}
	}printf("%d",f[T][0]); return 0;
}