POJ 3311 Hie with the Pie floyd+狀壓DP

鏈接：http://poj.org/problem?id=3311

題意：有N個地點和一個出發點(N<=10)，給出所有地點兩兩之間的距離，問從出發點出發，走遍所有地點再回到出發點的最短距離是多少。

思路：首先用floyd找到所有點之間的最短路。然後用狀態壓縮，dp數組一定是二維的，如果是一維的話不能保證dp[i]->dp[j]一定是最短的。因爲dp[i]記錄的“當前位置”不一定是能使dp[j]最小的當前位置。所以dp[i][j]中，i表示的二進制下的當前已經經過的狀態，j表示的是在當前狀態下目前所在的位置。

代碼：

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <ctype.h>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>
#define eps 1e-8
#define INF 0x7fffffff
#define PI acos(-1.0)
#define seed 31//131,1313
#define maxn 15
typedef long long LL;
typedef unsigned long long ULL;
using namespace std;
int dp[1<<10][maxn];
int Pow[maxn];
int back[maxn];
int cost[maxn][maxn];
void init1()
{
    Pow[0]=1;
    for(int i=1; i<=10; i++)
        Pow[i]=Pow[i-1]*2;
}
void init2()
{
    for(int i=1; i<(1<<10); i++)
        for(int j=0; j<=10; j++)
            dp[i][j]=INF;
}
int floyd(int a[][maxn],int t)
{
    for(int i=0; i<t; i++)
        for(int j=0; j<t; j++)
            for(int k=0; k<t; k++)
                a[i][j]=min(a[i][k]+a[k][j],a[i][j]);
}
int main()
{
    int T,x;
    init1();
    while(scanf("%d",&T))
    {
        init2();
        if(!T)
            break;
        for(int i=0; i<T+1; i++)
            for(int j=0; j<T+1; j++)
                scanf("%d",&cost[i][j]);
        floyd(cost,T+1);

        for(int i=0; i<T; i++)
            dp[Pow[i]][i]=cost[0][i+1];
        for(int i=0; i<T; i++)
            back[i]=cost[i+1][0];
        for(int i=0; i<T; i++)
            for(int j=0; j<T; j++)
                cost[i][j]=cost[i+1][j+1];
        for(int i=0; i<(1<<T); i++)
        {
            if(i==1||i==2||i==4||i==8||i==16||i==32||i==64||i==128||i==256||i==512)
            continue;
            int ii=i;
            int pos=0;
            while(ii)
            {
                if(ii%2==1)
                {
                    int t=i-Pow[pos];
                    for(int j=0; j<T; j++)
                        if(dp[t][j]!=INF&&dp[t][j]+cost[j][pos]<dp[i][pos])
                            dp[i][pos]=dp[t][j]+cost[j][pos];
                }
                ii>>=1;
                pos++;
            }
        }
        int ans=INF;
        for(int i=0; i<T; i++)
        {
            if(dp[(1<<T)-1][i]!=INF&&dp[(1<<T)-1][i]+back[i]<ans)
                ans=dp[(1<<T)-1][i]+back[i];
        }

        printf("%d\n",ans);
    }
    return 0;
}