UAa116 例题9-4 单向TSP

原题链接： UVa116

题目大意：

 给定一个n*m的矩阵，要求从第一列的任何一行出发，每次沿右或右下或右上到达下一列，图形的环形的，也就是说到达最后一行再向下走之后到达第一行，最后到第m列任何一行整个路程的最小值，并且要求是字典序最小的。

解题思路：

 本题是紫书上的例题9-4，看了题感觉有点思路就自己开始写，没有看分析，后来踩了很多坑，不过有些坑只有踩了以后才能避免。

 我的整体思路就是定义状态d[i][j]表示以（i,j）点为起点到最后一列的最短路径长度。状态转移方程就是d[i][j] = matrix[i][j]+min(d（右上）,d（右），d(右下))。输出时进行状态判断正确之后输出。具体细节参考代码

代码：

我的代码：

//E9-4 UVa116  AC 
#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int MAXN = 100 + 5;
const int MAXM = 10 + 5;
const int INF = 1 << 30;

int d[MAXM][MAXN], matrix[MAXM][MAXN], row[MAXN];
int m, n;

bool if_this(int r ,int c ,int last)//判断该列是不是取该
{
	int last_row = row[c - 1];
	return (r == ((last_row + m - 1) % m) || r == ((last_row + 1) % m) || r == last_row) && last == (d[r][c] + matrix[row[c - 1]][c - 1]);
}

void print_ans(int last)		//打印路径 ，last上一列的最对应的路径最小值 
{
	cout << row[0] + 1;
	for (int i = 1; i < n; i++) {
		for (int j = 0; j < m; j++) {
			if (if_this(j, i, last)) {//如果上一列的最小值等于该点的d值加上一列求得最小值所在的点的值，说明找到该列的取最小值的位置 
				row[i] = j;	last = d[j][i];
				cout << " " << row[i] + 1;
				break; 
			}
		}
	}
}

int dp(int r, int c)		//dp搜素 
{
	if (d[r][c] < INF ) return d[r][c];
	d[r][c] = matrix[r][c];	
	if (c < n-1) {				//判断是否为边界 
		int up = dp((r + m -1 ) % m , c + 1);//计算三个方向上的最小值 
		int down = dp((r + 1 )% m, c + 1);
		int mid = dp(r, c + 1);
		int min_d = min(up,mid );
		d[r][c] = d[r][c] + min(min_d, down);//取最小值 
	}
	return d[r][c];
}

int main()
{
	//freopen("input.txt","r",stdin);
	//freopen("output.txt","w",stdout);
	while (cin >> m >> n) {				//输入并初始化d 
		for (int i = 0; i < m; i++) {
			for (int j = 0; j < n; j++) {
				cin >> matrix[i][j];
				d[i][j] = INF;
			}
		}
		int min_len = INF;
		for (int i = 0; i < m; i++){	//对第一列的每个点进行dp，寻找最小值 
			if (dp(i, 0) < min_len) {
				min_len = dp(i, 0);
				row[0] = i;
			}
		}
		print_ans(min_len);		//打印路径 
		cout << endl << min_len << endl;
	}
	return 0;
}

书上的代码：

// UVa116 Unidirectional TSP
// Rujia Liu
// 算法：多段图的动态规划。因为要字典序最小，所以倒着来，设d[i][j]为从(i,j)到最后一列的最小开销，则d[i][j]=a[i][j]+max(d[i+1][j+1],d[i-1][j+1])
#include<cstdio>
#include<algorithm>
using namespace std;

const int maxn = 100 + 5;
const int INF = 1000000000;
int m, n, a[maxn][maxn], d[maxn][maxn], next[maxn][maxn];

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

    int ans = INF, first = 0;
    for(int j = n-1; j >= 0; j--) {
      for(int i = 0; i < m; i++) {
        if(j == n-1) d[i][j] = a[i][j];
        else {
          int rows[3] = {i, i-1, i+1};
          if(i == 0) rows[1] = m-1;
          if(i == m-1) rows[2] = 0;
          sort(rows, rows+3);
          d[i][j] = INF;
          for(int k = 0; k < 3; k++) {
            int v = d[rows[k]][j+1] + a[i][j];
            if(v < d[i][j]) { d[i][j] = v; next[i][j] = rows[k]; }
          }
        }
        if(j == 0 && d[i][j] < ans) { ans = d[i][j]; first = i; }
      }
    }
    printf("%d", first+1);
    for(int i = next[first][0], j = 1; j < n; i = next[i][j], j++) printf(" %d", i+1);
    printf("\n%d\n", ans);
  }
  return 0;
}