116 - Unidirectional TSP 【DP】

题目大意

传送门
给一个m行n列的整数矩阵，从第一列任何一个位置出发依次往右、右上、右下走一格，最终到达最后一列。
要求经过的整数之和最小。矩阵的第一行的上一行是最后一行，最后一行的下一行是第一行。
输出路径上每行的行号。多个解的时候输出字典序小的。

样例

input

5 6
3 4 1 2 8 6
6 1 8 2 7 4
5 9 3 9 9 5
8 4 1 3 2 6
3 7 2 8 6 4
5 6
3 4 1 2 8 6
6 1 8 2 7 4
5 9 3 9 9 5
8 4 1 3 2 6
3 7 2 1 2 3
2 2
9 10 9 10

output

1 2 3 4 4 5
16
1 2 1 5 4 5
11
1 1
19

解释

思路

d[i][j]表示为从格子(i,j)出发到最后一列的最小开销。
则d[x][n]都为edge[x][n]，这个是初始态。

对状态做初始化：

for (int i = 0; i < m; i++)
for (int j = 0; j < n; j++)
{
    if (j == n-1) dp[i][j] = edge[i][j];
    else dp[i][j] = INF;
}

递推公式：

for (int j = n - 2; j >= 0; j--)
    for (int i = 0; i < m; i++)
        for (int k = -1; k <= 1; k++)
        {
                dp[i][j] = min(dp[i][j], dp[(i+k+m)%m][j + 1] + edge[i][j]);
        }

代码

#include <stdio.h>
#define MAXN 105
#define INF 1<<30
#define min(a,b) (a<b)? a:b 

int edge[MAXN][MAXN];
int dp[MAXN][MAXN];
int m, n;


int main(){
    while (~scanf("%d %d",&m,&n))
    {
        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
                scanf("%d", &edge[i][j]);

        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++){
                if (j == n-1) dp[i][j] = edge[i][j];
                else dp[i][j] = INF;
            }


        for (int j = n - 2; j >= 0; j--)
        {
            for (int i = 0; i < m; i++)
            {
                for (int k = -1; k <= 1; k++)
                {

                    dp[i][j] = min(dp[i][j], dp[(i+k+m)%m][j + 1] + edge[i][j]);
                }
            }
        }

        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                printf("%d\t", dp[i][j]);
            }
            printf("\n");
        }
    }
}

Hit

在递推公式的时候
要注意的是 j 要在循环的最外面，算完了一组j之后才可以算j-1。
如果j的循环在第二层，那就错了