并查集,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;

}

相关推荐
Tisfy13 分钟前
LeetCode 2411.按位或最大的最小子数组长度:一次倒序遍历
数据结构·算法·leetcode·题解·位运算·遍历
2202_7567496940 分钟前
04 基于sklearn的机械学习-梯度下降(上)
人工智能·算法·机器学习
草莓爱芒果41 分钟前
Spring Boot中使用Bouncy Castle实现SM2国密算法(与前端JS加密交互)
java·spring boot·算法
晚云与城1 小时前
【数据结构】-----排序的艺术画卷
数据结构·算法·排序算法
weixin_307779131 小时前
设计Mock CUDA库的流程与实现
c++·算法·gpu算力
j_xxx404_1 小时前
数据结构:算法复杂度与空间复杂度
c语言·数据结构·算法
dlraba8022 小时前
基于 OpenCV 与 sklearn 的数字识别:KNN 算法实践
opencv·算法·sklearn
yzzzzzzzzzzzzzzzzz2 小时前
leetcode热题——全排列
算法·回溯·全排列
王柏龙2 小时前
mongodb中的哈希索引详解
算法·mongodb·哈希算法
NAGNIP2 小时前
GPT1:通用语言理解模型的开端
后端·算法