UVA116 Unidirectional TSP 单向TSP

#include<bits/stdc++.h>
using namespace std;

const int maxm = 11;
const int maxn = 101;

int G[maxm][maxn];
const int INF = 1e9;
int d[maxm][maxn];
int path[maxn][maxn];

int main()
{
//freopen("in.txt","r",stdin);
int m,n;
while(~scanf("%d%d",&m,&n)){
for(int i = 0; i < m; i++)
for(int j = 0; j < n ;j++){
scanf("%d",G[i]+j);
}
int ans = INF,head = 0;
for(int i = 0; i < m; i++) d[i][n-1] = G[i][n-1];
if(n == 1){
for(int i = 0; i < m; i++){
if(d[i][0]<ans) { ans = d[i][0]; head = i; }
}
}else {
for(int j = n-2; j >= 0; j--){
for(int i = 0; i < m; i++){
int nxt[] = {i,i+1,i-1};
if(i == 0) nxt[2] = m-1;
if(i == m-1) nxt[1] = 0;
sort(nxt,nxt+3);
d[i][j] = d[nxt[0]][j+1];
path[i][j] = nxt[0];
for(int k = 1; k < 3; k++){
int t = d[nxt[k]][j+1];
if(t < d[i][j]){
d[i][j] = t; path[i][j] = nxt[k];
}
}
d[i][j] += G[i][j];
if(j == 0 && d[i][j] < ans) { ans = d[i][j]; head = i; }
}
}
}

printf("%d",head+1);
for(int i = path[head][0],j = 1; j < n; i = path[i][j], j++){
printf(" %d",i+1);
}
printf("\n%d\n",ans);
}
return 0;
}