1. 程式人生 > >並查集(C++實現)

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