题目链接:
1013 Battle Over Cities - PAT (Advanced Level) Practice (pintia.cn)
题目大意:
题目要求我们在多个城市之间,如果某个城市被占领,所有连接这个城市的高速公路都将不可用。我们需要计算在占领某个城市之后,剩下的城市之间最少需要多少条高速公路才能保证所有剩余的城市依然连通。
通过并查集来确定该城市消失过后需要修复的公路的数量
并查集基本知识点
当某个城市 x
被占领时,我们忽略与这个城市相关的所有高速公路,判断剩下的城市是否还连通。如果不连通,则需要额外修复高速公路使其重新连通。
cnt
变量的作用是用于统计当前剩余需要连通的城市之间的高速公路数量。这个 cnt
表示连通操作中剩余的城市数量减去已经成功连通的部分,从而判断还需要多少条高速公路才能让所有剩余的城市连通起来。
当时,困住我的点是,如何跳过被摧毁的城市以及如何统计需要修复的公路。
我刚开始想的是通过循环f数组来统计集合的数量,答案是集合数量-1,但是发现还要排除被摧毁城市,有点玛法,所以可以通过用cnt来表示需要修复的公路,初始值为n - 1,一旦两个城市可以连通就cnt --即可
#include <bits/stdc++.h>
using namespace std;
const int N = 1010,M = 500010;
typedef struct{
int start,End;
}node;
int f[N];
node a[M];
int find(int x){
return f[x] = (f[x] == x ? x : find(f[x]));
}
int main() {
ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
int n,m,k; cin >> n >> m >> k;
for(int i = 0; i < m; i++) cin >> a[i].start >> a[i].End;
for(int i = 0; i < k; i++){
int x; cin >> x;
int cnt = n - 1;
// 初始化
for(int i = 1; i <= n; i++) f[i] = i;
for(int j = 0; j < m; j++){
int c1 = a[j].start, c2 = a[j].End;
if(c1 != x && c2 != x){
int pa = find(c1),pb = find(c2);
if(pa != pb){
f[pa] = pb;
cnt --;
}
}
}
cout << cnt - 1 <<'\n';
}
return 0;
}