1. 程式人生 > >Power收集[單調佇列+DP]

Power收集[單調佇列+DP]

傳送門

f[i][j]=f[i-1][x]+val[i][j](i-t<=x<=i+t)

將上一輪合法的加進來 , 更新答案之前彈出和重新將後面和發的放進來就可以了


 

#include<bits/stdc++.h>
#define N 4050
using namespace std;
int val[N][N],f[N][N];
int n,m,k,t,q[N],ans;
int main(){
	scanf("%d%d%d%d",&n,&m,&k,&t);
	for(int i=1;i<=k;i++){
		int x,y; scanf("%d%d",&x,&y);
		scanf("%d",&val[x][y]);
	}
	for(int i=1;i<=m;i++) f[1][i] = val[1][i];
	for(int i=2;i<=n;i++){
		
		int l=1,r=0;
		for(int j=1;j<=min(m,t);j++){
			while(l<=r && f[i-1][q[r]] <= f[i-1][j]) r--;
			q[++r] = j; 
		}
		
		for(int j=1;j<=m;j++){
			if(j+t <= m){
				while(l<=r && f[i-1][q[r]] <= f[i-1][j+t]) r--;
				q[++r] = j+t;
			}
			if(l<r && q[l]+t < j) l++;
			
			f[i][j] = f[i-1][q[l]] + val[i][j];
		}
	}
	for(int i=1;i<=m;i++) ans = max(ans , f[n][i]);
	printf("%d",ans); return 0;
}