并查集模板(Disjoint-set Data Structure)

//Disjoint-set
#include <cstdio>

//修改DS_size更改使用上界
const int DS_size=100000; 

//pre:表示x的父亲是pre[x]
int pre[DS_size+5];
//表示x属于的集合的大小。作用：①用于启发式策略参考;②对外提供size(i)查询。
int siz[DS_size+5];
//初始化
inline void init(int n){
    for(int i=1;i<=n;i++)siz[pre[i]=i]=1;
}
//查找x集合的代表节点，同时并查集
inline int find(int x){
    return x==pre[x]?x:pre[x]=find(pre[x]);
}
//求i属于的集合大小
inline int size(int x){
    return siz[find(x)];
}

//x,y属于的集合求交，小集合成为大集合的子树
void uni(int x,int y){
    if((x=find(x))!=(y=find(y))){
        if(siz[x]>siz[y])siz[pre[y]=x]+=siz[y];
        else siz[pre[x]=y]+=siz[x];
    }
}

//测试
int main(){
    int n;
    init(n=5);
    uni(1,2);
    for(int i=1;i<=n;i++)printf("%d ",size(i));
    puts("");
    uni(2,3);
    for(int i=1;i<=n;i++)printf("%d ",size(i));
    puts("");
    uni(4,5);
    for(int i=1;i<=n;i++)printf("%d ",size(i));
    puts("");
    uni(3,5);
    for(int i=1;i<=n;i++)printf("%d ",size(i));
    puts("");
    init(n);
    uni(3,5);
    for(int i=1;i<=n;i++)printf("%d ",size(i));
    puts("");
}