最小斯坦纳树

给一个图,边权为正,给一个点集,求一个子图,包含集合中所有点,联通,且边权和最小。点集大小不大

首先这肯定是个树,树肯定是联通了,且边权和最小。考虑反证法,如果不是树,删掉一个边,边权和肯定变小。

由于点集不大,考虑状压记录当前的子图里有多少个集合中的点。由于无根树,不同根的答案可能不同,考虑换根。

转移可以是我们考虑当前点集是由哪两个子集拼起来的,这是个子集枚举,也可以增加一条边,但点集不变,故有两个转移
在这里插入图片描述
子集的转移枚举子集即可,另一转移类似求最短路的松弛操作,所以我们可以对于当前点集的dp数组,真的跑一次最短路,把此时 d p [ m a s k ] dp[mask] dp[mask]里不为初始值的点都放入初始队列

void solve() {
    int n,m,k;
    cin>>n>>m>>k;
    vector<vvi>g(n);
    rep(i,1,m){
		int u,v,w;
		cin>>u>>v>>w;
		u--,v--;
		g[u].push_back({v,w});
		g[v].push_back({u,w});
	}
	vvi dp(1<<k,vi(n,1e9));
	rep(i,0,k-1){
		int x;
		cin>>x;
		dp[1<<i][x-1]=0;
	}
	auto spfa=[&](int s)->void{
		queue<int>q;
		vi vis(n);
		rep(i,0,n-1){
			if(dp[s][i]!=1e9){
				q.push(i);
				vis[i]=1;
			}
		}	
		while(q.size()){
			int u=q.front();
			q.pop();
			vis[u]=0;
			for(auto &t:g[u]){
				int v=t[0],w=t[1];
				if(dp[s][v]>dp[s][u]+w){
					dp[s][v]=dp[s][u]+w;
					if(!vis[v]){
						q.push(v);
						vis[v]=1;
					}
				}
			}
		}
	};
	rep(mask,1,(1<<k)-1){
		for(int sub=mask&(mask-1);sub;sub=(sub-1)&mask){
			rep(i,0,n-1){
				dp[mask][i]=min(dp[mask][i],dp[mask^sub][i]+dp[sub][i]);
			}
		}
		spfa(mask);
	}
	int ans=1e9;
	rep(i,0,n-1){
		//cout<<dp[(1<<k)-1][i]<<' ';
		ans=min(ans,dp[(1<<k)-1][i]);
	}
	cout<<ans;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值