题面:BZOJ1855 Luogu2569
DP肯定的
状态:f[i][j]表示前i天目前持有j股的最大收益
有几种转移方式:
不买不卖:f[i][j]=f[i-1][j]
买:f[i][j]=max(f[i-1][k]-ap[i]*(j-k))(k < j)
卖:f[i][j]=max(f[i-1][k]+bp[i]*(k-j))(k > j)
但是枚举k的话时间复杂度是O(t*maxP^2)无法接受
那么我们可以用单调队列维护k
时间复杂度就降到了O(t*maxP)
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <iostream>
#include <ctime>
#include <map>
#include <queue>
#include <cstdlib>
#include <string>
#include <climits>
#include <set>
#include <vector>
using namespace std;
int n,m,w,ap[2001],bp[2001],as[2001],bs[2001];
int f[2001][2001],q[100001];
int main()
{
scanf("%d%d%d",&n,&m,&w);
for(int i=1;i<=n;i++)scanf("%d%d%d%d",&ap[i],&bp[i],&as[i],&bs[i]);
memset(f,-0x7f,sizeof f);
for(int i=0;i<=n;i++)f[i][0]=0;
for(int i=1;i<=n;i++){
for(int j=0;j<=as[i];j++)f[i][j]=-ap[i]*j;
for(int j=0;j<=m;j++)f[i][j]=max(f[i][j],f[i-1][j]);
if(i-w-1>=0){
int l=1,r=0;
for(int j=0;j<=m;j++){
while(l<=r&&q[l]<j-as[i])l++;
while(l<=r&&f[i-w-1][j]+ap[i]*j>=f[i-w-1][q[r]]+ap[i]*q[r])r--;
q[++r]=j;f[i][j]=max(f[i][j],f[i-w-1][q[l]]-ap[i]*(j-q[l]));
}
l=1,r=0;
for(int j=m;j>=0;j--){
while(l<=r&&q[l]>j+bs[i])l++;
while(l<=r&&f[i-w-1][j]+bp[i]*j>=f[i-w-1][q[r]]+bp[i]*q[r])r--;
q[++r]=j;f[i][j]=max(f[i][j],f[i-w-1][q[l]]+bp[i]*(q[l]-j));
}
}
}
int ans=0;
for(int i=0;i<=m;i++)ans=max(ans,f[n][i]);
printf("%d",ans);
return 0;
}