CCF CSP认证201703-4 地铁修建（Kruskal最小生成树）

201703-4 地铁修建

题目

思路

这次的第四题比较简单，就是一个Kruskal最小生成树算法，我还一度担心“公司数为n”这块有个坑。

题目的要求是，找一条1至n的路径，使这条路径上的最大边在所有1至n路径最大边中最小。因为有n个公司，而1至n的无环路径边数肯定是小于n的，所以不用考虑一个公司修两条路的情况。

Kruskal算法可用并查集实现，将所有边按权排序，从小的开始枚举，如果边的两端不在一个集合内，就将其合并。当1和n刚好并至一个集合内时，最小生成树1至n路径上的最大边就是这个刚好使1和n合并的边。
可以这么理解，从小边开始枚举，找出使1与n连通所必需的最大边。

AC代码如下

#include<cstdio>
#include<algorithm>
using namespace std;
struct edge{//存边
 	int u,v,c;
 	friend bool operator < (edge a,edge b) {return a.c<b.c;}
}E[200005];
int n,m,f[100005];
int findf(int x){//并查集的根结点查找函数
 	if(f[x]==x) return x;
 	return f[x]=findf(f[x]);//路径压缩
}
int kruskal(){
 	for(int i=1;i<=n;i++) f[i]=i;
 	sort(E,E+m);//边排序
 	for(i=0;i<m;i++){//从小边开始枚举
  		int fa=findf(E[i].u);
  		int fb=findf(E[i].v);
  		if(fa!=fb){
   			f[fa]=fb;//合并
   			if(findf(1)==findf(n)) return E[i].c;//E[i]两端合并后，1与n连通则返回E[i]边长
  		}
 	}
}
int main(){
 	scanf("%d%d",&n,&m);
 	for(int i=0;i<m;i++)  scanf("%d%d%d",&E[i].u,&E[i].v,&E[i].c);
 	printf("%d",kruskal());
 	return 0;
}