Contest 2050 and Codeforces Round #718 (Div. 1 + Div. 2) D. Explorer Space最短路 floyd变种

题目链接

D. Explorer Space
time limit per test2 seconds
memory limit per test256 megabytes
inputstandard input
outputstandard output
You are wandering in the explorer space of the 2050 Conference.

The explorer space can be viewed as an undirected weighted grid graph with size n×m. The set of vertices is {(i,j)|1≤i≤n,1≤j≤m}. Two vertices (i1,j1) and (i2,j2) are connected by an edge if and only if |i1−i2|+|j1−j2|=1.

At each step, you can walk to any vertex connected by an edge with your current vertex. On each edge, there are some number of exhibits. Since you already know all the exhibits, whenever you go through an edge containing x exhibits, your boredness increases by x.

For each starting vertex (i,j), please answer the following question: What is the minimum possible boredness if you walk from (i,j) and goes back to it after exactly k steps?

You can use any edge for multiple times but the boredness on those edges are also counted for multiple times. At each step, you cannot stay on your current vertex. You also cannot change direction while going through an edge.

Input
The first line contains three integers n, m and k (2≤n,m≤500,1≤k≤20).

The j-th number (1≤j≤m−1) in the i-th line of the following n lines is the number of exibits on the edge between vertex (i,j) and vertex (i,j+1).

The j-th number (1≤j≤m) in the i-th line of the following n−1 lines is the number of exibits on the edge between vertex (i,j) and vertex (i+1,j).

The number of exhibits on each edge is an integer between 1 and 106.

Output
Output n lines with m numbers each. The j-th number in the i-th line, answerij, should be the minimum possible boredness if you walk from (i,j) and goes back to it after exactly k steps.

If you cannot goes back to vertex (i,j) after exactly k steps, answerij should be −1.

Examples
inputCopy
3 3 10
1 1
1 1
1 1
1 1 1
1 1 1
outputCopy
10 10 10
10 10 10
10 10 10
inputCopy
2 2 4
1
3
4 2
outputCopy
4 4
10 6
inputCopy
2 2 3
1
2
3 4
outputCopy
-1 -1
-1 -1
Note
In the first example, the answer is always 10 no matter how you walk.

In the second example, answer21=10, the path is (2,1)→(1,1)→(1,2)→(2,2)→(2,1), the boredness is 4+1+2+3=10.

代码

#include<bits/stdc++.h>
using namespace std;
// #define int long long
int n,m,k;
const int N=510;
int f[N][N][11];
int a[N][N];
int b[N][N];
void solve()
{
    cin>>n>>m>>k;
    if(k&1)
    {
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
                cout<<-1<<" ";
            cout<<'\n';
        }
        return ;
    }
    k>>=1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<m;j++)
            scanf("%d",a[i]+j);
    for(int i=1;i<n;i++)
        for(int j=1;j<=m;j++)
            scanf("%d",b[i]+j);
    for(int c=1;c<=k;c++)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
            {
                f[i][j][c]=1e9;
                if(i>1)f[i][j][c]=min(f[i-1][j][c-1]+b[i-1][j],f[i][j][c]);
                if(i<n)f[i][j][c]=min(f[i+1][j][c-1]+b[i][j],f[i][j][c]);
                if(j>1)f[i][j][c]=min(f[i][j-1][c-1]+a[i][j-1],f[i][j][c]);
                if(j<m)f[i][j][c]=min(f[i][j+1][c-1]+a[i][j],f[i][j][c]);
            }
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
            cout<<f[i][j][k]*2<<' ';
        cout<<'\n';
    }
}
signed main()
{
    solve();
    return 0;
}