这个例子是从洛谷的 P1536 村村通 中我的题解中截取的,删掉了一个while循环,变成了普遍的并查集模板写法,再次记录以防以后忘记。
#include <iostream>
#include <vector>
using namespace std;
#define endl '\n'
#define int long long
int n, m;
int find(int x, vector<int>& fa) {
return fa[x] == x ? x : (fa[x] = find(fa[x], fa));
}//也可以用全局变量fa
signed main() {
cin >> n >> m;//n个点,m条边
int k = n;
vector<int> fa(n + 1);
for (int i = 1; i <= n; i++) {
fa[i] = i;
}
int x, y;
for (int i = 0; i < m; i++) {
cin >> x >> y;
int fx = find(x, fa);
int fy = find(y, fa);
if (fx != fy) {
fa[fx] = fy;
n--;
}
}
//这里循环得到的n,是并过之后有n个连通块
for (int i = 1; i < fa.size(); i++) {
find(i, fa);
}
//这里是把并查集完全合并
//所有连通块的的元素的fa都指向同一个
cout << n - 1 << endl;//输出还要几条边才能使图完全连通
return 0;
}