并查集(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;
}
 

猜你喜欢

转载自blog.csdn.net/h471507602/article/details/81088877
今日推荐