【NOIP practice】BSOJ 2998修复公路 最小生成树

2988 -- 【基础试题】修复公路
Description
　　A地区在地震过后，连接所有村庄的公路都造成了损坏而无法通车。政府派人修复这些公路。
　　给出A地区的村庄数N，和公路数M，公路是双向的。并告诉你每条公路的连着哪两个村庄，并告诉你什么时候能修完这条公路。问最早什么时候任意两个村庄能够通车，即最早什么时候任意两条村庄都存在至少一条修复完成的道路（可以由多条公路连成一条道路）
Input
　　第1行两个正整数N，M（N<=1000，M<=100000）
　　下面M行，每行3个正整数x, y, t，告诉你这条公路连着x,y两个村庄，在时间t时能修复完成这条公路。（x<=N，y<=N，t<=100000）
Output
　　如果全部公路修复完毕仍然存在两个村庄无法通车，则输出-1，否则输出最早什么时候任意两个村庄能够通车。
Sample Input
4 4
1 2 6
1 3 4
1 4 5
4 2 3
Sample Output
5

这是一道纯粹的最小生成树。

为什么呢？将时间看做边权，要在一个图里面找出让每个点连接的最小方案——最小生成树（“且包含原图中的所有 n 个结点，并且有保持图连通的最少的边”）。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
struct node
{
	int fr,to,val;
}w[500005];
int n,m;
int h[500005]={0},cnt=0,ans=0;
int prt[500005]={0};
void Addarc(int x,int y,int v)
{
	cnt++;w[cnt].fr=x;w[cnt].to=y;w[cnt].val=v;
}
bool cmp(node x,node y)
{
	return x.val<y.val;
}
void init()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++)
	{
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		Addarc(a,b,c);
		Addarc(b,a,c);
	}
}
int Getfa(int x)
{
	if(x==prt[x])return x;
	return prt[x]=Getfa(prt[x]); 
}
void K()
{
	int ct=0;
	for(int i=1;i<=n;i++)prt[i]=i;
	for(int i=1;i<=2*m;i++)
	{
		int f1=Getfa(w[i].fr),f2=Getfa(w[i].to);
		if(f1!=f2)
		{
			prt[f1]=f2;
			ans=max(ans,w[i].val);
			ct++;
		}
		if(ct==n-1)break;
	}
	if(ct==n-1)printf("%d\n",ans);
	else printf("-1\n");
}
int main(){

	init();
	sort(w+1,w+2*m+1,cmp);//按边权排序，单向边，双向边，共2m条边 
	K();
	return 0;
}