题目链接:Problem - 1517D - Codeforces Explorer Space
在一个无向图中,求每一个点走k步回到起点的最短路径,允许走重复的路,不难想,k如果是奇数的,肯定是无法回到原点的,只要每个点计算走k/2的步数的最短路路径即可。数据量比较小,考虑dp,dp[i][j][k]表示从(i,j)这个点出走k步所到达的最小权值。这道题还有一点需要说明,就是图的存储方式比较特殊。
#include<bits/stdc++.h> using namespace std; int n,m,k,g[505][505][4],f[505][505][12]; int main(){ scanf("%d%d%d",&n,&m,&k); for(int i=1;i<=n;i++) for(int j=1;j<=m-1;j++){ scanf("%d",&g[i][j][0]); g[i][j+1][2]=g[i][j][0]; } for(int i=1;i<=n-1;i++) for(int j=1;j<=m;j++){ scanf("%d",&g[i][j][1]); g[i+1][j][3]=g[i][j][1]; } if(k&1){ for(int i=1;i<=n;i++){ for(int j=1;j<=m;j++){ if(j!=m) printf("-1 "); else printf("-1\n"); } } return 0; } for(int t=1;t<=k/2;t++) for(int i=1;i<=n;i++) for(int j=1;j<=m;j++){ int res=0x3f3f3f3f;//边界初始化为0,边界内部初始化无穷大,这样写也是可行的 if(i>1){ res=min(res,f[i-1][j][t-1]+g[i-1][j][1]); } if(i<n){ res=min(res,f[i+1][j][t-1]+g[i+1][j][3]); } if(j>1){ res=min(res,f[i][j-1][t-1]+g[i][j-1][0]); } if(j<m){ res=min(res,f[i][j+1][t-1]+g[i][j+1][2]); } f[i][j][t]=res; } for(int i=1;i<=n;i++){ for(int j=1;j<=m;j++){ if(j!=m) printf("%d ",f[i][j][k/2]*2); else printf("%d\n",f[i][j][k/2]*2); } } return 0; }