HDU - 3001 Travelling （三进制状压）

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

题意：给一个图，求走遍所有点的最短路径，每个点最多可以走2次。

思路：预处理出三进制下每种状态第i个城市走了几次=>mk[sta][i]，然后初始化，开始可以到任何城市，所以把每个城市dp[只到了第i个城市的状态][i]=0,其他为INF，dp[sta][i]就是sta状态下最后到达第i个城市的最短路径。

状态更新就是 dp[sta+只到到第j个城市的状态][j]=min(dp[sta+只到到第j个城市的状态][j],dp[sta][i]+dis[i][j])。

相当于以第i个城市为跳板，走到第j个城市来更新状态（类似floyd），最后当这种状态每个城市都走了至少一次，那么就可以来更新答案。

注意：有重边！！！

代码：

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define eps 1e-8
#define fuck(x) cout<<"<"<<x<<">"<<endl
#define fi first
#define se second
#define pb push_back 
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
using namespace std;
typedef long long LL;
typedef unsigned long long uLL;
typedef pair<int, int> pii;
const double PI = acos(-1.0);
const LL INFLL = 0x3f3f3f3f3f3f3f3fll;
const int maxn = 5e6 + 5;
const int mod = 1e9 + 7;


int n,m;
int d[15][15],b[15],mk[60005][15];
int dp[60005][15];
int main() {
    b[0]=1;
    for (int i=1;i<=10;i++) b[i]=b[i-1]*3;
    for (int i=0;i<b[10];i++){
        int tmp=i;
        for (int j=0;j<10;j++){
            mk[i][j]=tmp%3;
            tmp/=3;
        }
    }
    while (~scanf ("%d%d",&n,&m)){
        memset(d,INF,sizeof(d));
        memset(dp,INF,sizeof(dp));
        for (int i=0;i<m;i++){
            int u,v,c;
            scanf ("%d%d%d",&u,&v,&c);
            if (d[u][v]>c) d[u][v]=d[v][u]=c;
        }
        for (int i=1;i<=n;i++) dp[b[i-1]][i]=0;
        int ans=INF;
        for (int sta=0;sta<b[n];sta++){
            bool flag=1;
            for (int i=1;i<=n;i++){
                if (mk[sta][i-1]==0) {flag=0;continue;}
                for (int j=1;j<=n;j++){
                    if (i==j) continue;
                    if (mk[sta][j-1]<2) dp[sta+b[j-1]][j]=min(dp[sta+b[j-1]][j],dp[sta][i]+d[i][j]);
                }
            }
            if (flag) for (int i=1;i<=n;i++) ans=min(ans,dp[sta][i]);
        }
        if (ans==INF) printf ("-1\n");
        else printf ("%d\n",ans);
    }
    return 0;
}