hdu 4640(状压dp）

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

思路：f[i][j]表示一个人状态i下走到j的最小花费，dp[i][j]表示i个人在状态j下的最下花费。首先我们可以一遍bfs求出f[i][j],然后通过f[i][j]得到dp[1][i],最后就是更新dp[i][j]了。

 1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<queue> 6 using namespace std; 7 #define inf 1<<30 8 typedef pair<int,int>PP; 9 10 int map[20][20];11 int f[1<<18][20];//派一个人去，在某状态下到达点v的最小花费12 int dp[4][1<<18];//派i个人去，在某状态的最小花费13 bool mark[1<<18][20];14 int n,m,S,ans;15 16 void bfs()17 {18     queue<PP>que;19     que.push(make_pair(0,0));20     memset(mark,false,sizeof(mark));21     for(int i=0;i<(1<<n);i++)22         for(int j=0;j<n;j++)f[i][j]=inf;23     f[0][0]=0;24     while(!que.empty()){25         PP p=que.front();26         que.pop();27         int s=p.first,u=p.second;28         mark[s][u]=false;29         for(int i=0;i<n;i++){30             if(map[u][i]<inf&&f[s|(1<<i)][i]>f[s][u]+map[u][i]){31                 f[s|(1<<i)][i]=f[s][u]+map[u][i];32                 if(!mark[s|(1<<i)][i]){33                     mark[s|(1<<i)][i]=true;34                     que.push(make_pair(s|(1<<i),i));35                 }36             }37         }38     }39 }40 41 void Solve()42 {43     for(int i=1;i<=3;i++)44         for(int j=0;j<(1<<n);j++)dp[i][j]=inf;45     for(int i=0;i<(1<<n);i++)46         for(int j=0;j<n;j++)dp[1][i]=min(dp[1][i],f[i][j]);47     for(int i=2;i<=3;i++){48         for(int j=0;j<(1<<n);j++){49             //枚举子集50             for(int k=j;k;k=(k-1)&j){51                 dp[i][j]=min(dp[i][j],max(dp[1][k|1],dp[i-1][(j^k)|1]));52             }53         }54     }55     ans=inf;56     for(int i=1;i<=3;i++)57         for(int j=0;j<(1<<n);j++)if((j&S)==S)ans=min(ans,dp[i][j]);58     if(ans==inf)ans=-1;59     printf("%d\n",ans);60 }61 62 int main()63 {64     int _case,u,v,w,k,t=1;65     scanf("%d",&_case);66     while(_case--){67         scanf("%d%d",&n,&m);68         for(int i=0;i<=n;i++)69             for(int j=0;j<=n;j++)map[i][j]=(i==j?0:inf);70         while(m--){71             scanf("%d%d%d",&u,&v,&w);72             u--,v--;73             map[u][v]=map[v][u]=min(map[u][v],w);74         }75         S=0;76         scanf("%d",&k);77         while(k--){78             scanf("%d",&u);79             u--;80             S|=(1<<u);81         }82         bfs();83         printf("Case %d: ",t++);84         Solve();85     }86     return 0;87 }88 89 90 91             

View Code