#include<stdio.h>
#define maxSize 10
//用森林的双亲表示法的思想,表示并查集
int S[maxSize];
//初始化,将每个元素的前驱下标设置为-1
void initial(int s[]){
for(int i=0;i<maxSize;i++){
s[i]=-1;
}
}
//实现"查"的操作,查这个元素所属的根节点的下标
int find(int s[],int k){
int root=k;
while(s[root]>=0) root=s[root];//找到根节点
//压缩路径,将k到根节点的所有结点都变为根节点的直接后继
while(k!=root){
int t=s[k];//指向k的父节点
s[k]=root;
k=t;
}
return root;
}
//实现"并"的操作,小树并到大树里面
void Union(int s[],int root1,int root2){
if(root1==root2){
return;
}
//确保root1和root2是根节点
if(s[root1]>=0){
root1=find(s,root1);
}
if(s[root2]>=0){
root2=find(s,root2);
}
//小树合并到大树里面
if(s[root1]>s[root2]){
s[root2]+=s[root1];//更新结点个数,每个根节点的数组值为连接的结点个数
s[root1]=root2;//合并
}else{
s[root1]+=s[root2];
s[root2]=root1;
}
}
int main(){
//自己的测试数据
S[0]=-1;S[1]=0;S[2]=-1;S[3]=-1;S[4]=1;S[5]=1;S[6]=2;S[7]=3;S[8]=3;S[9]=3;S[10]=4;
printf("5的根节点是%d\n",find(S,5));
printf("6的根节点是%d\n",find(S,6));
//合并
Union(S,0,2);
printf("5的根节点是%d\n",find(S,5));
printf("6的根节点是%d\n",find(S,6));
return 0;
}