HDU 3401 Trade （单调队列优化DP）

来源：互联网发布：ssr centos 一键编辑：程序博客网时间：2024/05/16 12:01
http://acm.hdu.edu.cn/showproblem.php?pid=3401
#include<iostream>#include<cstdio>#include<cstring>#include<queue>using namespace std;#define N 2399#define inf 0xfffffffstruct data{    int v,k;    bool operator<(data a)const { return v>a.v; }    data(int a=0,int b=0) { v=a,k=b; }}q[N];int n,maxp,W,AP[N],BP[N],AS[N],BS[N];int dp[N][N];int main(){    int re; cin>>re;    while(re--) {        cin>>n>>maxp>>W;        for(int i=1;i<=n;i++)            scanf("%d%d%d%d",AP+i,BP+i,AS+i,BS+i);        for(int i=0;i<=n;i++)            for(int j=0;j<=maxp;j++)            dp[i][j]=-inf;        dp[0][0]=0;      ///  priority_queue<data> q[N],qq[N];        for(int i=1;i<=W+1;i++)            for(int j=0;j<=min(AS[i],maxp);j++)            dp[i][j]=-AP[i]*j;        for(int i=2;i<=n;i++)            for(int j=0;j<=maxp;j++)            dp[i][j]=max(dp[i][j],dp[i-1][j]);        for(int i=W+2 ;i<=n;i++) {            for(int j=0;j<=maxp;j++) ;            if(i<=W+1) continue;            int pre=i-W-1;            ///buy            int head=0,tail=-1;            for(int j=0;j<=maxp;j++)            {                dp[i][j]=max(dp[i][j],dp[i-1][j]);                data tmp=data(dp[i-W-1][j]+j*AP[i],j);               // data tmp=q[i-W-1].top();  q[i-W-1].pop();                while(head<=tail&& q[tail].v<tmp.v )  tail--;                q[++tail]=tmp;                while(head<=tail&& j-q[head].k > AS[i]) head++;                if(head<=tail)                    dp[i][j]=max(dp[i][j],  q[head].v -j*AP[i]);            }            ///sell            head=0, tail=-1;            for(int j=maxp;j>=0;j--)            {                dp[i][j]=max( dp[i][j], dp[i-1][j] );                data tmp=data( dp[pre][j]+ j*BP[i], j );                ///data tmp= qq[i-W-1].top();  qq[i-W-1].pop();                while(head<=tail && q[tail].v<tmp.v ) tail--;                q[++tail]=tmp;                while(head<=tail&& q[head].k-j>BS[i] ) head++;                if(head<=tail)                    dp[i][j]=max(dp[i][j] , q[head].v - j*BP[i] );            }        }        int ans=0;        for(int j=0;j<=maxp;j++)            ans=max(ans,dp[n][j]);        cout<<ans<<endl;    }}
0 0