一道很不错的优先队列题
f[i][j]的最大值由上一行的状态有关,转移方程f[i][j]=max(f[i-1][z])+a[i][j],z∈(j-k,j+k)
每个转移会超时,所以我们对上一行建立单调队列,由1推到m,提前入队,超出范围出队。
#include<iomanip>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m,k,t,f[4001][4001],a[4001][4001];
int main()
{
cin>>n>>m>>k>>t;
memset(a,0,sizeof(a));
for(int i=1;i<=k;++i)
{
int x,y,z;
cin>>x>>y>>z;
a[x][y]=z;
}
for(int i=1;i<=n;++i)
for(int j=1;j<=m;++j)
f[i][j]=-99999;
for(int i=1;i<=m;++i)
f[0][i]=0;
for(int i=1;i<=n;++i)
{
int q[4001],head=1,tail=0;
for(int j=1;j<=t;++j)
{
while(f[i-1][q[tail]]<=f[i-1][j]&&head<=tail) tail--;
q[++tail]=j;
}
for(int j=1;j<=m;++j)
{
if(q[head]<j-t&&head<=tail) head++;
if(j+t<=m)
{
while(f[i-1][q[tail]]<=f[i-1][j+t]&&head<=tail) tail--;
q[++tail]=j+t;
}
f[i][j]=f[i-1][q[head]]+a[i][j];
}
}
int maxx=0;
for(int i=1;i<=m;++i)
if(f[n][i]>maxx) maxx=f[n][i];
cout<<maxx;
return 0;
}