CCF CSP 20181204 数据中心【瓶颈生成树】

题目描述

题目链接 CCF CSP 20181204

题意分析

• 对于给定无向图，求解一颗生成树，使得最大边权最小
• 即求解最小瓶颈生成树

所用性质

• 最小生成树一定是瓶颈生成树

瓶颈生成树——百度百科
【证明】：假设最小生成树不是瓶颈树，设最小生成树T的最大权边为e，则存在一棵瓶颈树Tb，其所有的边的权值小于w(e)。删除T中的e，形成两棵树T’, T’’，用Tb中连接T’, T’'的边连接这两棵树，得到新的生成树，其权值小于T，与T是最小生成树矛盾。

• 利用Kruskal求解最小生成树

数据结构

采用并查集。

• 利用par[]数组记录节点i的代表元节点。
• 初始化：将所有节点的parp[]值设置为本身。
• 查找操作：利用路径压缩，丢弃节点关系，直接把自己挂在根节点下
• 合并操作：如果两个元素分属不同的并查集，把一个分组的根挂到另一个分组的根即可。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=5e4+10;
struct Edge{
	int u,v,w;
	bool operator<(const Edge &e) const{
		return w<e.w;
	}
};
Edge Es[2*maxn];
int par[maxn];
int tot,ans;
//并查集
void init(int n){
	for(int i=1;i<=n;i++){
		par[i]=i;
	}
}
int find(int x){
	return par[x]==x?x:par[x]=find(par[x]);
}
bool unite(int x,int y){
	x=find(x),y=find(y);
	if(x==y) return false;
	par[x]=y;
	return true;
}
int Kruskal(int n){
	int cnt=0;
	init(n);
	sort(Es,Es+tot);
	for(int i=0;i<tot;i++){
		if(unite(Es[i].u,Es[i].v)){
			if(++cnt==n-1){
				return Es[i].w;
			}
		}
	}
}
int main(){
	int n,m,root;cin>>n>>m>>root;
	for(int i=1;i<=m;i++){
		cin>>Es[tot].u>>Es[tot].v>>Es[tot].w;
		tot++;
	}
	cout<<Kruskal(n)<<endl;
	return 0;
	
}