洛谷 P1111 修复公路 并查集

https://www.luogu.org/problem/P1111
题目背景
A地区在地震过后，连接所有村庄的公路都造成了损坏而无法通车。政府派人修复这些公路。
题目描述
给出A地区的村庄数N，和公路数M，公路是双向的。并告诉你每条公路的连着哪两个村庄，并告诉你什么时候能修完这条公路。问最早什么时候任意两个村庄能够通车，即最早什么时候任意两条村庄都存在至少一条修复完成的道路（可以由多条公路连成一条道路）
输入格式
第1行两个正整数N,M
下面M行，每行3个正整数x,y,t，告诉你这条公路连着x,y两个村庄，在时间t时能修复完成这条公路。
输出格式
如果全部公路修复完毕仍然存在两个村庄无法通车，则输出−1，否则输出最早什么时候任意两个村庄能够通车。

思路：并查集水题，加一个$siz$数组就行了，把道路按照时间从小到大排序然后再处理。

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define eps 1e-8
#define pr pair<int,int>
using namespace std;
typedef long long ll;

const int maxn=1005;
const int maxm=1e5+5;

struct node
{
    int u,v,t;
    node(){}
    node(int a,int b,int c)
    {
        u=a,v=b,t=c;
    }
    bool operator <(const node &a)const
    {
        return t<a.t;
    }
}a[maxm];

int f[maxn],siz[maxn];
int n,m;

void init()
{
    for(int i=1;i<=n;i++)
        f[i]=i,siz[i]=1;
}

int father(int x)
{
    return f[x]==x?x:f[x]=father(f[x]);
}

int main()
{
    scanf("%d %d",&n,&m);
    init();
    int u,v,t;
    for(int i=0;i<m;i++)
    {
        scanf("%d %d %d",&u,&v,&t);
        a[i]=node(u,v,t);
    }
    sort(a,a+m);
    bool flag=0;
    for(int i=0;i<m;i++)
    {
        int fx=father(a[i].u);
        int fy=father(a[i].v);
        if(fx!=fy)
        {
            f[fx]=fy;
            siz[fy]+=siz[fx];
            if(siz[fy]==n)
            {
                flag=1;
                printf("%d\n",a[i].t);
                break;
            }
        }
    }
    if(!flag)
        printf("-1\n");
    return 0;
}