修复公路
题目背景
A 地区在地震过后,连接所有村庄的公路都造成了损坏而无法通车。政府派人修复这些公路。
题目描述
给出 A 地区的村庄数 N N N,和公路数 M M M,公路是双向的。并告诉你每条公路的连着哪两个村庄,并告诉你什么时候能修完这条公路。问最早什么时候任意两个村庄能够通车,即最早什么时候任意两条村庄都存在至少一条修复完成的道路(可以由多条公路连成一条道路)。
输入格式
第 1 1 1 行两个正整数 N , M N,M N,M。
下面 M M M 行,每行 3 3 3 个正整数 x , y , t x,y,t x,y,t,告诉你这条公路连着 x , y x,y x,y 两个村庄,在时间t时能修复完成这条公路。
输出格式
如果全部公路修复完毕仍然存在两个村庄无法通车,则输出 − 1 -1 −1,否则输出最早什么时候任意两个村庄能够通车。
样例 #1
样例输入 #1
4 4
1 2 6
1 3 4
1 4 5
4 2 3
样例输出 #1
5
数据规模与约定
1 ≤ x , y ≤ N ≤ 1 0 3 1\leq x, y\leq N \le 10 ^ 3 1≤x,y≤N≤103, 1 ≤ M , t ≤ 1 0 5 1\leq M, t \le 10 ^ 5 1≤M,t≤105。
原题
思路
按时间 t t t 对所有公路的修建进行排序,然后按时间顺序模拟修复公路的过程(即并查集的合并操作),一旦并查集中某个集合的大小达到 n n n (即在某一次合并操作过后所有集合最终合并为一个集合),那么就说明此时任意两个村庄能够通车,输出该时刻即可。而当遍历完所有公路的修复后仍没有某个集合的大小达到 n n n,则说明存在两个村庄无法通车,输出 − 1 -1 −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;
}