历届试题 国王的烦恼
时间限制:1.0s 内存限制:256.0MB
问题描述
C国由n个小岛组成,为了方便小岛之间联络,C国在小岛间建立了m座大桥,每座大桥连接两座小岛。两个小岛间可能存在多座桥连接。然而,由于海水冲刷,有一些大桥面临着不能使用的危险。
如果两个小岛间的所有大桥都不能使用,则这两座小岛就不能直接到达了。然而,只要这两座小岛的居民能通过其他的桥或者其他的小岛互相到达,他们就会安然无事。但是,如果前一天两个小岛之间还有方法可以到达,后一天却不能到达了,居民们就会一起抗议。
现在C国的国王已经知道了每座桥能使用的天数,超过这个天数就不能使用了。现在他想知道居民们会有多少天进行抗议。
输入格式
输入的第一行包含两个整数n, m,分别表示小岛的个数和桥的数量。
接下来m行,每行三个整数a, b, t,分别表示该座桥连接a号和b号两个小岛,能使用t天。小岛的编号从1开始递增。
输出格式
输出一个整数,表示居民们会抗议的天数。
样例输入
4 4
1 2 2
1 3 2
2 3 1
3 4 3
样例输出
2
样例说明
第一天后2和3之间的桥不能使用,不影响。
第二天后1和2之间,以及1和3之间的桥不能使用,居民们会抗议。
第三天后3和4之间的桥不能使用,居民们会抗议。
数据规模和约定
对于30%的数据,1<=n<=20,1<=m<=100;
对于50%的数据,1<=n<=500,1<=m<=10000;
对于100%的数据,1<=n<=10000,1<=m<=100000,1<=a, b<=n, 1<=t<=100000。
分析:用遍历的办法会超时。用并查集反向构筑可以在规定时间内求出结果。
具体思路如下:将所有桥的起点终点及期限按照期限长短降序排列,然后从期限最长的桥(即最后被淹没的桥)开始遍历所有桥,使用并查集判断桥的起点与终点之间是否已有通路。若已有通路,则说明不会引起抗议,否则有抗议。此时查看当前的日期是否为第一次出现,是则说明此天发生暴动,天数加一,否则说明此天已经发生过暴动,无需天数加一。
代码:
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
struct Node {
int u;
int v;
int t;
};
bool cmp(Node a, Node b) {
return a.t > b.t;
}
int n, m;
vector<Node> vn;
vector<int> Father;
int findFather(int x) {
if (Father[x] == x) {
return x;
}
return Father[x] = findFather(Father[x]);
}
int isUnion(int a, int b) {
int faA = findFather(a);
int faB = findFather(b);
if (faA != faB) {
Father[faB] = faA;
return 0;
}
return 1;
}
int main() {
cin >> n >> m;
vn.resize(m);
Father.resize(n + 1);
for (int i = 0; i < m; i++) {
cin >> vn[i].u >> vn[i].v >> vn[i].t;
}
sort(vn.begin(), vn.end(), cmp);
for (int i = 1; i <= n; i++) {
Father[i] = i;
}
int ans = 0;
int pre = -1;
for (int i = 0; i < m; i++) {
int u = vn[i].u;
int v = vn[i].v;
int t = vn[i].t;
if (!isUnion(u, v) && t != pre) {
pre = vn[i].t;
ans++;
}
}
cout << ans << endl;
return 0;
}