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

#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
;

int n,maxp,W,
AP
,BP
,AS
,BS
;
int dp

;
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
,qq
;
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
[j]);
cout<<ans<<endl;
}
}