洛谷 P1111 修复公路题解并查集

最新推荐文章于 2024-07-19 21:09:58 发布

bbc_ouo

最新推荐文章于 2024-07-19 21:09:58 发布

阅读量569

点赞数 11

分类专栏：洛谷题解数据结构文章标签： c++ 算法

本文链接：https://blog.csdn.net/bbc_plus/article/details/138390659

版权

洛谷题解同时被 2 个专栏收录

51 篇文章 0 订阅

订阅专栏

数据结构

16 篇文章 0 订阅

订阅专栏

文章介绍了利用并查集解决公路修复问题，计算最早公路修复完成后所有村庄连通的时间，若无法完全连通则输出-1。

摘要由CSDN通过智能技术生成

修复公路

题目背景

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

题目描述

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

输入格式

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

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

输出格式

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

样例 #1

样例输入 #1

样例输出 #1

数据规模与约定

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

原题

洛谷P1111——传送门

思路

按时间 $t$ 对所有公路的修建进行排序，然后按时间顺序模拟修复公路的过程(即并查集的合并操作)，一旦并查集中某个集合的大小达到 $n$ (即在某一次合并操作过后所有集合最终合并为一个集合)，那么就说明此时任意两个村庄能够通车，输出该时刻即可。而当遍历完所有公路的修复后仍没有某个集合的大小达到 $n$ ，则说明存在两个村庄无法通车，输出 $- 1$ 。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

struct DSU
{
    std::vector<int> f, siz;

    DSU() {}
    DSU(int n)
    {
        init(n);
    }

    void init(int n)
    {
        f.resize(n);
        std::iota(f.begin(), f.end(), 0);
        siz.assign(n, 1);
    }

    int find(int x)
    {
        while (x != f[x])
        {
            x = f[x] = f[f[x]];
        }
        return x;
    }

    bool same(int x, int y)
    {
        return find(x) == find(y);
    }

    bool merge(int x, int y)
    {
        x = find(x);
        y = find(y);
        if (x == y)
        {
            return false;
        }
        siz[x] += siz[y];
        f[y] = x;
        return true;
    }

    int size(int x)
    {
        return siz[find(x)];
    }
};

struct operation // 公路修复的结构体
{
    int x, y, t; // t时刻在x和y之间修复公路
};

bool cmp(operation a, operation b) // 按t排序
{
    return a.t < b.t;
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    int n, m;
    cin >> n >> m;
    DSU dsu = DSU(n);
    vector<operation> v(m);
    for (int i = 0; i < m; i++)
    {
        cin >> v[i].x >> v[i].y >> v[i].t;
        // 并查集结构体中下标从0开始，题目从1开始，所以通过--操作来对应下标
        v[i].x--;
        v[i].y--;
    }
    sort(v.begin(), v.end(), cmp); // 按t排序
    for (int i = 0; i < m; i++)
    {
        dsu.merge(v[i].x, v[i].y); // 合并
        if (dsu.size(v[i].x) == n) // 该次合并操作后集合大小达到n，则任意两个村庄可通车
        {
            cout << v[i].t;
            return 0;
        }
    }
    cout << -1; // 无法连通所有村庄，输出-1

    return 0;
}