洛谷P1111 修复公路（并查集）

一.题目 # P1111 修复公路

## 题目背景

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

## 题目描述

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

## 输入格式

第 $1$ 行两个正整数 $N,M$。

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

## 输出格式

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

## 输入输出样例 #1

### 输入 #1

```
4 4
1 2 6
1 3 4
1 4 5
4 2 3
```

### 输出 #1

```
5
```

## 说明/提示

$1\leq x, y\leq N \le 10 ^ 3$，$1\leq M, t \le 10 ^ 5$。

二.思路

很明显，这个题目要求的就是把所有的点拼接成树然后判断是否能够联通，所以这里使用并查集就可以。

使用并查集的细节，首先进行路径压缩后直接合并就好了，如果合并后有两个集合，说明失败了

三.代码

#include<algorithm>
#include <iostream>
#include<vector>
using namespace std;
vector<int>parent;
int N, M;
struct node
{
    int x, y, t;
}a[100000 + 10];
int find(int x)//路径压缩
{
    if (parent[x] != x) {
        parent[x] = find(parent[x]);//寻找父节点
    };
    return parent[x];//返回找到的父节点
}

void m(int x, int y)
{
    int rootx = find(x);
    int rooty = find(y);
    if (rootx != rooty)
    {
        parent[rootx] = rooty;
    }
}
bool cmp(node fir, node sec)
{
    return fir.t < sec.t;
} 
int check()
{
    int sum = 0;
    for (int i = 1; i <= N; i++)
    {
        if (parent[i] == i)sum++;
        if (sum == 2) return 0;

    }
    return 1;
}

int main()
{
    cin >> N >> M;
    parent.resize(N + 1);
    for (int i = 1; i <=N; i++)parent[i] = i;
    for (int i = 1; i <= M; i++)cin >> a[i].x >> a[i].y >> a[i].t;
    sort(a + 1, a + M + 1,cmp );
    for (int i = 1; i <= M; i++)
    {
        m(a[i].x, a[i].y);
        if (check())//如果只有1个集合
        {
            printf("%d\n", a[i].t);//输出时间
            return 0;//愉快的结束主程序
        }
    }
    cout << -1 << endl;
    return 0;
}