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

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 0xfffffff
struct 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;
    }
}