【并查集】洛谷 P1111 修复公路

题目背景

A地区在地震过后，连接所有村庄的公路都造成了损坏而无法通车。政府派人修复这些公路。

题目描述

给出A地区的村庄数N，和公路数M，公路是双向的。并告诉你每条公路的连着哪两个村庄，并告诉你什么时候能修完这条公路。问最早什么时候任意两个村庄能够通车，即最早什么时候任意两条村庄都存在至少一条修复完成的道路（可以由多条公路连成一条道路）

输入输出格式

输入格式：
第1行两个正整数N，M

下面M行，每行3个正整数x, y, t，告诉你这条公路连着x,y两个村庄，在时间t时能修复完成这条公路。

输出格式：
如果全部公路修复完毕仍然存在两个村庄无法通车，则输出-1，否则输出最早什么时候任意两个村庄能够通车。

输入输出样例

输入样例#1：
4 4
1 2 6
1 3 4
1 4 5
4 2 3
输出样例#1：
5

说明

N<=1000，M<=100000

x<=N，y<=N，t<=100000

代码

#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN=1000+10,MAXM=100000+10;
int fa[MAXN];
struct node{
    int x;
    int y;
    int t;
};
node a[MAXM];
int comp(const node&i,const node&j)
{
    return i.t<j.t;
}
int found(int x)
{
    if(fa[x]!=x)fa[x]=found(fa[x]);
    return fa[x];
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=m;i++)scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].t);
    sort(a+1,a+m+1,comp);
    int k=0;
    for(int i=1;i<=m;i++)
    {
        if(found(a[i].x)!=found(a[i].y))
        {
            fa[found(a[i].x)]=found(a[i].y);
            k++;
        }
        if(k==n-1){cout<<a[i].t;break;}
    }
    if(k<n-1)cout<<-1;
    return 0;
}