Explorer Space

题目链接：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;
} 
​