并查集,find函数的压缩路径,Union函数的小树合并大树

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

}

相关推荐
程序趣谈3 小时前
算法随笔_74: 不同路径_1
数据结构·python·算法
修修修也4 小时前
算法手记3
数据结构·学习·算法·刷题
吧啦吧啦吡叭卜5 小时前
【打卡d5】快速排序 归并排序
java·算法·排序算法
L_cl5 小时前
【Python 数据结构 15.哈希表】
数据结构·算法·散列表
肖筱小瀟6 小时前
2025-3-13 leetcode刷题情况(贪心算法--区间问题)
算法·leetcode·贪心算法
肖筱小瀟6 小时前
2025-3-12 leetcode刷题情况(贪心算法--区间问题)
算法·leetcode·贪心算法
小王努力学编程7 小时前
元音辅音字符串计数leetcode3305,3306
开发语言·c++·学习·算法·leetcode
不忘不弃7 小时前
矩阵的转置
线性代数·算法·矩阵