股票交易[單調佇列優化DP]
阿新 • • 發佈:2018-11-19
我們用f[i][j] 表示交易到第i天,手上有j個股票的最大收益
對於買入
變一下形
單調佇列維護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; }