Codeforces Round #718 (Div. 1 + Div. 2) D 棋盘步数回路 DP 动态规划 思维

题目

给你一个大小N*M的棋盘网格，网格的每条线都有一条无向边，以及它的权值。
要求
求每个点出发走K步后回到起点的最小权值和。

题解思路

当K为奇数是必然回不到起点。特判。
因为它是回路，我们只需求走k/2步的最小值再*2即可。
然后就是DP状态定义
F[i][x][y]
从x，y出发走了i步的最小权值和。
如何转移？
因为无向边，所以我们走k步最小到达的点，必然我们可以反向的到达回去。
相当于将状态等价为
F[i][x][y]
从未知点走了i步后到达x,y最小权值和。（这里很思维）

这样我们就能很轻松的写出如下的转移方程。

f[x][y][i] = 1e18 ; 
if ( x > 1 )
   f[x][y][i] = min(f[x][y][i],f[x-1][y][i-1] + w2[x-1][y]) ;
if ( y > 1 )
   f[x][y][i] = min(f[x][y][i],f[x][y-1][i-1] + w1[x][y-1]) ;
if ( x < n )
   f[x][y][i] = min(f[x][y][i],f[x+1][y][i-1] + w2[x][y]) ; 
if ( y < m )
   f[x][y][i] = min(f[x][y][i],f[x][y+1][i-1] + w1[x][y]) ;

AC代码

/*从你的全世界路过.*/
#include <bits/stdc++.h>
//#include <unordered_map>
//priority_queue
#define PII pair<int,int>
#define ll long long

using namespace std;

const  int  INF =  0x3f3f3f3f;
const  int  N =  200100;
int w1[510][510] ;
int w2[510][510] ; 
long long  f[510][510][20] ; 
int n , m , k ; 
void solve()
{
    cin >> n >> m >> k ;
    for (int i = 1 ; i <= n ; i++ )
        for (int j = 1 ; j < m ; j++ )
            cin >> w1[i][j] ; 
    for (int i = 1 ; i < n ; i++ )
        for (int j = 1 ; j <= m ; j++ )
            cin >> w2[i][j] ; 
    if ( k % 2)
    {
        for (int i = 1 ; i <= n ; i++ )
        {
            for (int j = 1 ; j <= m ; j++ )
                cout << "-1 " ; 
            cout << "\n" ; 
        }
        return ; 
    }
    for (int i = 1 ; i <= k/2 ; i++ )
    {
        for (int x = 1 ; x <= n ; x++ )
            for (int y = 1 ; y <= m ; y++ )
            {
                f[x][y][i] = 1e18 ; 
                if ( x > 1 )
                    f[x][y][i] = min(f[x][y][i],f[x-1][y][i-1] + w2[x-1][y]) ;
                if ( y > 1 )
                    f[x][y][i] = min(f[x][y][i],f[x][y-1][i-1] + w1[x][y-1]) ;
                if ( x < n )
                    f[x][y][i] = min(f[x][y][i],f[x+1][y][i-1] + w2[x][y]) ; 
                if ( y < m )
                    f[x][y][i] = min(f[x][y][i],f[x][y+1][i-1] + w1[x][y]) ;
            }
    }
    for (int i = 1 ; i <= n ; i++ )
    {
        for (int j = 1 ; j <= m ; j++ )
            cout << f[i][j][k/2]*2 << " " ;
        cout << "\n" ; 
    }
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
        solve() ;
    return 0 ;
}