HDU-3001 Travelling

题目：

http://acm.hdu.edu.cn/showproblem.php?pid=3001

题意：

TSP问题，最多10个点，给出之间的路线长度，求遍历的最短总路径长。

思路：

状压DP，注意是每个点不能拜访超过两次，所以是三进制状压。

dp[i+pos[k]][k] = min(dp[i+pos[k]][k], dp[i][j]+g[j][k]);

dp[i][j]表示在i状态时当前位置是j的最短路径。

代码：

#define N 11
#define M 112345

int n,m;
int g[N][N],pos[N];
int dp[M][N],vis[M][N];
void init()
{
    int i,j,t;
    pos[0]=1;
    for(i=1;i<N;i++)
        pos[i] = pos[i-1]*3;
    for(i=0;i<=pos[N-1];i++)
        for(t=i,j=0;j<N;j++)
            vis[i][j]=t%3,t/=3;
}
int solve()
{
    int ans=INF;
    for(int i=0;i<n;i++)
        dp[pos[i]][i] = 0;
    for(int i=0;i<pos[n];i++)
    {
        int flag=1;
        for(int j=0;j<n;j++)
        {
            if(vis[i][j]==0)flag=0;
            if(dp[i][j]==INF)continue;
            for(int k=0;k<n;k++)if(j!=k)
            {
                if(vis[i][k]==2||g[j][k]==INF)continue;
                dp[i+pos[k]][k] = min(dp[i+pos[k]][k],dp[i][j]+g[j][k]);
            }
        }
        if(flag)
            for(int j=0;j<n;j++)
                ans=min(ans,dp[i][j]);
    }
    if(ans==INF)ans=-1;
    return ans;
}
int main()
{
    int i,j,k,kk,t,x,y,z;
    init();
    while(scanf("%d%d",&n,&m)!=EOF&&n)
    {
        memset(g,0x3f,sizeof(g));
        memset(dp,0x3f,sizeof(dp));
        for(i=0;i<m;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            x--;y--;
            g[x][y]=g[y][x]=min(g[x][y],z);
        }
        printf("%d\n",solve());
    }
    return 0;
}