首页 > 代码库 > wikioi-1748 瑰丽华尔兹 -单调队列优化DP
wikioi-1748 瑰丽华尔兹 -单调队列优化DP
根据题意,很明显可以推出DP方程。
假如只考虑向左的方向:
dp[t][i][j]: 第t个时间段末滑行到i,j最长滑行的距离。
dp[t][i][j]=dp[t-1][i][1..k]+(j-k)=dp[t-1][i][1..k]-k+j(k<=j)很明显,可以使用单调队列优化。
最终时间复杂度为O(n*m*k)
#include<stdio.h> #include<string.h> #include<algorithm> #include<iostream> #include<queue> using namespace std; #define INF ((1<<30)-1) #define maxn 220 #define LL long long #define MOD 1000000009 struct list { int x; int val; }node[maxn+maxn],p; int dp[maxn][maxn][maxn]; int maps[maxn][maxn]; int main() { int i,j,m,n,x,y,k,st,ed,f,t; char str[1100]; while(~scanf("%d%d%d%d%d",&n,&m,&x,&y,&k)) { for(i=1;i<=n;i++) { scanf("%s",str); for(j=1;j<=m;j++) { if(str[j-1]==‘.‘)maps[i][j]=1; else maps[i][j]=0; } } for(i=0;i<=n;i++) for(j=0;j<=m;j++) for(t=0;t<=k;t++)dp[t][i][j]=-INF; int head,tail; dp[0][x][y]=0; for(t=1;t<=k;t++) { scanf("%d%d%d",&st,&ed,&f); int ti=ed-st+1; if(f==1){ for(j=1;j<=m;j++) { head=1;tail=0; for(i=n;i>=1;i--) { if(maps[i][j]){ p.val=dp[t-1][i][j]+i; p.x=i; while(tail>=head&&p.val>=node[tail].val)tail--; node[++tail]=p; while(tail>=head&&node[head].x-i>ti)head++; dp[t][i][j]=max(dp[t][i][j],node[head].val-i); } else{ head=1,tail=0; dp[t][i][j]=-INF; } } } } if(f==2){ for(j=1;j<=m;j++) { head=1;tail=0; for(i=1;i<=n;i++) { if(maps[i][j]){ p.val=dp[t-1][i][j]-i; p.x=i; while(tail>=head&&p.val>=node[tail].val)tail--; node[++tail]=p; while(tail>=head&&i-node[head].x>ti)head++; dp[t][i][j]=max(dp[t][i][j],node[head].val+i); } else{ head=1,tail=0; dp[t][i][j]=-INF; } } } } if(f==3){ for(i=1;i<=n;i++) { head=1;tail=0; for(j=m;j>=1;j--) { if(maps[i][j]){ p.val=dp[t-1][i][j]+j; p.x=j; while(tail>=head&&p.val>=node[tail].val)tail--; node[++tail]=p; while(tail>=head&&node[head].x-j>ti)head++; dp[t][i][j]=max(dp[t][i][j],node[head].val-j); } else{ head=1,tail=0; dp[t][i][j]=-INF; } } } } if(f==4){ for(i=1;i<=n;i++) { head=1;tail=0; for(j=1;j<=m;j++) { if(maps[i][j]){ p.val=dp[t-1][i][j]-j; p.x=j; while(tail>=head&&p.val>=node[tail].val)tail--; node[++tail]=p; while(tail>=head&&j-node[head].x>ti)head++; dp[t][i][j]=max(dp[t][i][j],node[head].val+j); } else{ head=1,tail=0; dp[t][i][j]=-INF; } } } } for(i=1;i<=n;i++) { for(j=1;j<=m;j++) { // if(dp[t][i][j]>=0)printf("%2d ",dp[t][i][j]); /// else printf("%2d ",-1); } // cout<<endl; } } int maxx=-1; for(t=1;t<=k;t++) { for(i=1;i<=n;i++) { for(j=1;j<=m;j++)maxx=max(maxx,dp[t][i][j]); } } cout<<maxx<<endl; } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。