hdu5418状压dp

g[i][j]:i 和 j  之间的最优解,通过 Floyd 算法求解

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=20,maxs=(1<<16)+5;
int n,m,g[maxn][maxn],c[maxs],dp[maxs][maxn];
//dp[s][i]:s 为曾到达过的城市的下标的映射,i是最后一个到达的城市的下标
int main(){
	int t,cas=0,S,u,v,w;
	for(int i=0;i<16;i++) c[1<<i]=i+1;
	scanf("%d",&t);
	while(t--){
		memset(g,inf,sizeof(g));
		memset(dp,inf,sizeof(dp));
		scanf("%d%d",&n,&m);
		S=(1<<n)-1;
		for(int i=1;i<=n;i++) g[i][i]=0;
		while(m--){
			scanf("%d%d%d",&u,&v,&w);
			g[u][v]=g[v][u]=min(g[v][u],w);
		}
		for(int k=1;k<=n;k++)
			for(int i=1;i<=n;i++)
				for(int j=1;j<n;j++)
					g[i][j]=min(g[i][j],g[i][k]+g[k][j]);
		dp[1][1]=0;	
		//重点:3for 
		for(int s=1;s<S;s++){
			for(int lst,i=s;i>0;i&=i-1){
				lst=c[-i&i];//遍历最后一位1 
				if(dp[s][lst]==inf) continue; //没出现过,就跳过	
                //lst,nex分别为上次和这次的结尾;s,cs分别为上次和这次共出现的 
		        //s和S^s关于S互补。所以下面这个for遍历剩余的未选的数
				for(int cs,nex,d=S^s;d>0;d&=d-1){
					nex=c[-d&d],cs=s|(-d&d);//也是按最后一位1的顺序遍历
					dp[cs][nex]=min(dp[cs][nex],dp[s][lst]+g[lst][nex]);
				}
			}
		}
		//dp[S][i]:S表示每个城市都访问过,i表示最后经过哪个城市回到起点 
		for(int i=2;i<=n;i++)
			dp[S][1]=min(dp[S][1],dp[S][i]+g[i][1]);
		printf("%d\n",dp[S][1]);
	}
}

猜你喜欢

转载自blog.csdn.net/weixin_50904510/article/details/120089229
今日推荐