TSP问题（动态规划求解）HDU5067

问题描述：

有n个城市，从第一个城市出发，每个城市只能走一次，再回到第一个城市，求最小代价。

思路：使用动态规划，dp[i][j]，其中i是压缩的状态，表示第i个城市是否已经走过，j表示第j个城市。dp[i][j]表示状态是i且当前所在的城市是j时最小代价。转移方程：

dp[i|(1<<k)][k] = min(dp[i|(1<<k)][k], dp[i][j]+dis[j][k])

即j城市已经走过且是目前所在的城市，k是尚未走过将要走的下一个城市，从j城市走到k城市。

代码：

memset(dp, INF, sizeof(dp));
dp[1][0] = 0;
for(int i = 1; i < (1<<cnt); i++){
    for(int j = 0; j < cnt; j++){  // j -> k
        if(!((1<<j) & i) || dp[i][j]==INF) continue;  // 未走过j 或 此路不通
        for(int k = 0; k < cnt; k++){
            if(((1<<k)&i) || (dis[j][k]==INF)) continue;// 已走过k 或者 j k不通
            dp[i|(1<<k)][k] = min(dp[i|(1<<k)][k], dp[i][j]+dis[j][k]);
        }
    }
}
int ans = INF;
for(int i = 0; i < cnt; i++) ans = min(ans, dp[(1<<cnt)-1][i] + dis[i][0]);
cout<<ans<<endl;

其中第14行是回路情况。

如果只是要求通路，即不要求再回到第一个城市，那么14行改成：

for(int i = 0; i < cnt; i++) ans = min(ans, dp[(1<<cnt)-1][i]);

练习题：HDU5067

代码：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
struct node{
    int to, len;
};
#define INF 0x3f3f3f3f
const int N = 12;
int dp[1<<(N-1)][N];
int a[55][55];
int dis[N][N];
int main(){
    int n, m;
    while(cin>>n>>m){
        int x[11], y[11];
        int cnt = 0;
        for(int i = 0; i < n; i++)
            for(int j = 0; j < m; j++){
                cin>>a[i][j];
                if(a[i][j] || i==0&&j==0){
                    x[cnt] = i;
                    y[cnt++] = j;
                }
            }
        for(int i = 0; i < cnt; i++) // 建图
            for(int j = i+1; j < cnt; j++){
                int len = abs(x[i]-x[j])+abs(y[i]-y[j]);
                dis[i][j] = dis[j][i] = len;
            }
        memset(dp, INF, sizeof(dp));
        dp[1][0] = 0;
        for(int i = 1; i < (1<<cnt); i++){
            for(int j = 0; j < cnt; j++){  // j -> k
                if(!((1<<j) & i) || dp[i][j]==INF) continue;  // 未走过j 或 此路不通
                for(int k = 0; k < cnt; k++){
                    if(((1<<k)&i) || (dis[j][k]==INF)) continue;// 已走过k 或者 j k不通
                    dp[i|(1<<k)][k] = min(dp[i|(1<<k)][k], dp[i][j]+dis[j][k]);
                }
            }
        }
        int ans = INF;
        for(int i = 0; i < cnt; i++) ans = min(ans, dp[(1<<cnt)-1][i] + dis[i][0]);
        cout<<ans<<endl;
    }
    return 0;
}