并查集(C++实现)

直接上代码,操作、解释都在里面’

#include 

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;
}

你可能感兴趣的:(图论)