并查集模板

貌似这么短的代码也不需要什么模板。。但还是写出来装下逼吧。。

带路径压缩的查找算法，需要对p[i]初始化为i，1<=i<=n，n是集合元素个数

int find(int x){
int r,t,i;
r=x;
while(r!=p[r])
r=p[r];
i=x;
while(i!=r){
t=p[i];
p[i]=r;
i=t;
}
return r;
}

按秩合并算法，需要对rank[i]初始化为0，1<=i<=n，n是集合元素个数，还要注意，该函数的使用前提是ra!=rb

void merge(int ra,int rb){
if(rank[ra]==rank[rb]){
p[rb]=ra;
rank[ra]++;
}else{
if(rank[ra]>rank[rb])
p[rb]=ra;
else
p[ra]=rb;
}
}

初始化

void init(int n){
int i;
for(i=1;i<=n;i++)//默认编号是1到n，n是集合元素个数
p[i]=i,rank[i]=0;
}