Travelling HDU-3001（3进制-状压dp）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3001；

题目大意：有n(n<=10)个城市，m条路联通着这些城市，你可以选择任意一个城市为起点开始你的旅行，一个城市最多只能去两次，问你如果要走完这n个城市最短的路径。

题目思路：因为n很小，所以可以用状压dp来解决这个问题。而一个城市最多只能去两次，所以我们可以选择用3进制来表示每一个点在当前状态已经被走过多少次了。

dp[i][state]表示在state这个状态下走到 i 点所要走的路径长度，那么我们就可以推出状态转移方程为：

dp[j][state+3^j] = min(dp[k][state]+dis[k][j])(k为state状态下所有已经到达过的点)。

具体实现看代码。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int,int>pii;
const int MX = 1e5+8;
const int INF = 0x3f3f3f3f;

int n,m;
int dis[11][11],th[11];
int dp[11][MX];
int pos[11];

int get_num(int state){
    //得到state状态下有多少个点已经走过以及到达的次数。
    int res = 0;
    for(int i = 0;i < n;i++){
        pos[i] = state%3;
        state/=3;
        if(pos[i]) res++;
    }
    return res;
}

int main(){
    //freopen("in.txt","r",stdin);
    th[0] = 1;
    for(int i = 1;i <= 10;i++) th[i] = 3*th[i-1];
    while(~scanf("%d%d",&n,&m)){
        memset(dp,0x3f,sizeof(dp));
        memset(dis,0x3f,sizeof(dis));
        for(int i = 0;i < m;i++){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            u--;v--;
            //题目给的点是从1开始的，我们枚举状压状态时是从0开始的，记得-1；
            dis[u][v] = min(dis[u][v],w);
            dis[v][u] = dis[u][v];
        }
        int ans = INF;
        for(int i = 1;i < th[n];i++){
            int res = get_num(i);
            for(int j = 0;j < n;j++){
                if(pos[j]){
                    if(res == 1)
                        dp[j][i] = 0;
                    if(dp[j][i] == INF) continue;
                    if(res == n){
                        //当n个点都走过的时候就更新答案；
                        ans = min(ans,dp[j][i]);
                    }
                    for(int k = 0;k < n;k++){
                        if(j != k && pos[k] < 2 && dis[j][k] != INF){
                            int nxt = i + th[k];
                            dp[k][nxt] = min(dp[k][nxt],dp[j][i]+dis[j][k]);
                        }
                    }
                }
            }
        }
        if(ans == INF) puts("-1");
        else printf("%d\n",ans);
    }
    return 0;
}