並查集(C++實現)
#include <iostream>
using namespace std;
const int MAX_N = 200001;
int par[MAX_N];
int rank_[MAX_N];
void init(int n){ // 初始化 n 個元素
for(int i = 0; i < n; ++i){
par[i] = i;
rank_[i] = 0;
}
}
int find_(int x){ // 尋找樹的根
if(par[x] == x){
return x;
}else{
return par[x] = find_(par[x]); // 直接連向祖先結點, 查詢的時候可以省時間
}
}
void unite(int x, int y){ // 合併 x 和 y 所在的集合
x = find_(x); // 尋找祖先
y = find_(y); // 尋找祖先
if(x == y) return ;
if(rank_[x] < rank_[y]){ // 短的接在長的後面
par[x] = y; // 把 x 的集合與 y的集合併攏,並把 y 當做祖先結點
}else{
par[y] = x;
if(rank_[x] == rank_[y]) rank_[x]++;
}
}
bool same(int x, int y){
return find_(x) == find_(y);
}
int main(){
solve();
return 0;
}