poj3311 狀態壓縮dp

比較經典的狀態壓縮dp,題目說了可以重複的走點,而且到一個點的最短路可能是通過其他的點間接到達,所以首先用Floyd處理一下,然後dp就可以了
狀態轉移公式

dp[i | (1<<(k-1))][k] = min(dp[i | (1<<(k-1))][k],dp[i][j]+mp[j][k]);

表示在狀態i的時候,到達目的j的最短路最後掃一遍就可以了
代碼:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<string>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<cstring>

using namespace std;
#define LL long long
vector<int>vec[100005];
int in[100005];
priority_queue<int>que;
int ans[100005];
int main()
{
    int n;
    while(scanf("%d",&n) && n!=0)
    {
        int mp[15][15];
        n++;
        for(int i = 0;i<n;i++)
        {
            for(int j = 0;j<n;j++)
            {
                scanf("%d",&mp[i][j]);
            }
        }
        for(int k = 0;k<n;k++)
        {
            for(int i = 0;i<n;i++)
            {
                if(i==k)continue;
                for(int j = 0;j<n;j++)
                {
                    if(i==j)continue;
                    if(mp[i][j]>mp[i][k] + mp[k][j])
                    {
                        mp[i][j] = mp[i][k] + mp[k][j];
                    }
                }
            }
        }
        int dp[1<<11][15];
        memset(dp,0x3f,sizeof(dp));
        for(int i = 1;i<n;i++)
        {
            dp[1<<(i-1)][i] = mp[0][i];
        }
        for(int i = 1;i<(1<<(n-1));i++)
        {
            for(int j = 1;j<n;j++)
            {
                if(!(i & (1<<(j-1))))continue;
                for(int k = 1;k<n;k++)
                {
                    dp[i | (1<<(k-1))][k] = min(dp[i | (1<<(k-1))][k],dp[i][j]+mp[j][k]);
                }
            }
        }
        int ans = 999999999;
        int all = (1<<(n-1))-1;
        for(int i = 1;i<n;i++)
        {
            ans = min(ans,dp[all][i]+mp[i][0]); // 從最短地方返回原點
        }
        printf("%d\n",ans);
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章