斯坦纳树

斯坦纳树问题是一种组合优化问题,大概就是正权无向图上连接k个指定点所需要的最小权值。这是洛谷的板子题

一句话题意:n个点m条边带权无向图选k个点连接,求最小权值。n100,m500,k10

首先我们最后连接出来的图一定是一棵树。然后看这个k的范围很难不想到状压。好那就状压。

dp[i][j]为以i节点为根,包含点集为j的最小代价,然后我们就变成了一个长的像个树形dp的东西(虽然并不是)。那么我们考虑状态的转移。我们发现有两个转移柿子:

  1. 对于集合s,我们枚举它的所有子集t,那么就有:

dp[i][s]=min(dp[i][t]+dp[i][st])

这部分的复杂度根据二项式定理是ni=1kCki2i=n(2+1)k=n3k的。其实就相当与把以 i 为根的两棵子树加起来。
2. 对于节点i,对所有其它节点j有转移:

dp[i][s]=min(dp[j][s]+w[i][j])

我们用已经包含在树中的点 j 来扩展未在树中的点 i ,相当于把连接这两个点的边加入树中,此时 i 被扩展到树内。

看起来长的就像个最短路,于是我们写个dijkstra解决。

于是这个问题就结束了。不懂的可以看看注释。

void dijkstra(int s){//dijkstra的板子 加上个集合s 
	memset(v,false,sizeof(v));
	while(!q.empty()){
		int x=q.top().x;q.pop();
		if(!v[x]){
			v[x]=true;
			for(int i=head[x];i;i=edge[i].next){
				if(dp[edge[i].v][s]>dp[x][s]+edge[i].w){
					dp[edge[i].v][s]=dp[x][s]+edge[i].w;
					g.x=edge[i].v;g.w=dp[edge[i].v][s];
					q.push(g);
				}
			}
		}
	}
}
int main(){
	memset(dp,0x3f,sizeof(dp));
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=m;i++){
		int u,v,w;scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);add(v,u,w);
	}
	for(int i=1;i<=k;i++){
		scanf("%d",&p[i]);dp[p[i]][1<<(i-1)]=0;
	}
	for(int s=1;s<(1<<k);s++){//枚举集合s 
		for(int i=1;i<=n;i++){//枚举点i 
			for(int j=s&(s-1);j;j=s&(j-1)){//枚举s的子集j 
				dp[i][s]=min(dp[i][s],dp[i][j]+dp[i][s^j]);//进行转移 
			}
			if(dp[i][s]!=0x3f3f3f3f){
				g.x=i;g.w=dp[i][s];q.push(g);
			}
		}
		dijkstra(s);//每次转移完一个集合的所有子集之后跑最短路 
	}
	printf("%d",dp[p[1]][(1<<k)-1]);//统计答案 实际上不管是p[1]还是p[k]什么的答案都一样 
}
posted @   gtm1514  阅读(70)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 使用C#创建一个MCP客户端
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示