UVa116 单向TSP 动态规划

题意：给出一个行列的整数矩阵，从第一列任何一个位置出发每次往右、右上或右下走一格，最终到达最后一列。要求经过的整数之和最小。整个矩形是环形，即第一行的上一行是最后一行，最后一行的下一行是第一行。输出路径上每列的行号。多解时输出字典序最小的。

思路：表示从格子出发到最后一列的最小开销。同时记录路径。

 

#include <stdio.h>
#include <algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;
int a[12][105],dp[12][105],Next[12][105];
int main(void)
{
    int i,j,m,n,ans,first,k;
    while(scanf("%d %d",&m,&n) != EOF) {
        for(i = 0; i < m; i++) {
            for(j = 0; j < n; j++) {
                scanf("%d",&a[i][j]);
            }
        }
        ans = INF,first = 0;
        for(j = n - 1; j >= 0; j--) {
            for(i = 0; i < m; i++) {
                if(j == n - 1)
                    dp[i][j] = a[i][j];
                else {
                    int rows[3] = {i - 1,i,i + 1};
                    if(i == 0) rows[0] = m - 1;
                    if(i == m - 1) rows[2] = 0;
                    sort(rows,rows + 3);
                    dp[i][j] = INF;
                    for(k = 0; k < 3; k++) {
                        int v = dp[rows[k]][j + 1] + a[i][j];
                        if(dp[i][j] > v) {
                            dp[i][j] = v;
                            Next[i][j] = rows[k];
                        }
                    }
                }
                if(j == 0 && dp[i][j] < ans) {
                    ans = dp[i][j];
                    first = i;
                }
            }

        }
        printf("%d",first + 1);
        for(i = Next[first][0],j = 1; j < n; i = Next[i][j], j++)
            printf(" %d",i + 1);
        printf("\n%d\n",ans);
    }
    return 0;
}