P1111 修复公路(并查集+排序)

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

我的思路

此题运用到排序+并查集，根据要求最短时间内使任意两村能够通车，所以需要按时间从短到长排序。再根据题意，任意两条村庄都存在至少一条修复完成的道路（可以由多条公路连成一条道路），可以很快想到并查集。

AC代码

#include<bits/stdc++.h>
using namespace std;
int n, m, f[1001];
struct str {
  int x, y, t;
}a[100001];
bool cmp(str b, str c) {
  return b.t < c.t;
}
int find(int x) {
  return f[x] == x ? x : f[x] = find(f[x]);//并查集查找 
}
int main() {
  cin >> n >> m;
  for (int i = 1; i <= m; i++)
    cin >> a[i].x >> a[i].y >> a[i].t;
  sort(a + 1, a + 1 + m, cmp);//按时间从短到长排序 
  for (int i = 1; i <= n; i++)//并查集需要用到的初始化 
    f[i] = i;
  for (int i = 1; i <= m; i++) {
    if (find(a[i].x) != find(a[i].y)) {//如果两村不通，则连通两村 
      f[find(a[i].x)] = find(a[i].y);
      n--;
    } 
    if (n == 1) {
      cout << a[i].t;
      return 0;
    }
  }
  cout << -1;
  return 0;
}