洛谷 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 <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cmath>

#define N 100005

struct NOTE
{
    int s,e,t;
}a[N];

int n,m;

int father[N];

bool cmp(NOTE a,NOTE b)
{
    return a.t<b.t;
}

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

int main()
{
    scanf ("%d%d",&n,&m);
    for (int i = 1; i <= m; i++)
    {
        scanf ("%d%d%d",&a[i].s,&a[i].e,&a[i].t);
        father[i]=i;
    }

    std::sort (a+1,a+m+1,cmp);

    int cnt=n;
    for (int i = 1; i <= m; i++)
    {
        int x = find(a[i].s);
        int y = find(a[i].e);
        if (x == y)
            continue;
        father[y]=x;
        cnt--;
        if (cnt == 1)
        {
            printf ("%d\n",a[i].t);
            return 0;
        }
    }
    printf("-1\n");
}