hdu--5418Victor and World+狀態壓縮DP

acm_lkl發表於2020-04-04

題目連結:點選進入
昨天晚上比賽的時候,我居然想著用dfs做這道題,果斷超時了。其實因為城市的數量非常的小,n最大才12,所以我們是可以藉助於二進位制進行狀態壓縮的;定義dp[s][i]表示當前訪問城市的情況為s時下一個訪問城市為i時的最小油量,則轉移方程為dp[s|(1<<(i-1))]=min(dp[s][j]+dis[i][j]),狀態轉移方程的意思就是從s中找到已經訪問過的城市j,然後再從j訪問i;dis[i][j]表示的是i,j之間最短路的長度。ans=min(dp[(1<

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;

const int INF=100000000+100;

int dis[20][20];
int dp[1<<20][20];
int n,m;

void floyd()
{
    for(int k=1;k<=n;k++)
      for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
           dis[i][j]=dis[i][j]<dis[i][k]+dis[k][j]?dis[i][j]:dis[i][k]+dis[k][j];
}

int main()
{
     int t;
     freopen("in.txt","r",stdin);
     scanf("%d",&t);
     while(t--)
     {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
           for(int j=1;j<=n;j++)
              dis[i][j]=INF;
           dis[i][i]=0;
        }
         for(int i=0;i<m;i++)
         {
             int a,b,w;
             scanf("%d%d%d",&a,&b,&w);
             if(dis[a][b]>w)
                dis[a][b]=dis[b][a]=w;
         }
         floyd();
         memset(dp,0x3f,sizeof(dp));
         dp[1][1]=0;
         for(int s=1;s<(1<<n);s++)
            for(int i=1;i<=n;i++)
              for(int j=1;j<=n;j++)
              {
                 if(!(s&(1<<(i-1)))&&(s&(1<<(j-1))))
                     dp[s|(1<<(i-1))][i]=min(dp[s|(1<<(i-1))][i],dp[s][j]+dis[j][i]);
              }
         int ans=0x3f3f3f3f;
         for(int i=1;i<=n;i++)
            ans=min(ans,dp[(1<<n)-1][i]+dis[i][1]);
         printf("%d\n",ans);
     }
  return 0;
}

相關文章