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

}

相关推荐
千金裘换酒24 分钟前
LeetCode 移动零元素 快慢指针
算法·leetcode·职场和发展
wm104344 分钟前
机器学习第二讲 KNN算法
人工智能·算法·机器学习
NAGNIP1 小时前
一文搞懂机器学习线性代数基础知识!
算法
NAGNIP1 小时前
机器学习入门概述一览
算法
iuu_star1 小时前
C语言数据结构-顺序查找、折半查找
c语言·数据结构·算法
Yzzz-F2 小时前
P1558 色板游戏 [线段树 + 二进制状态压缩 + 懒标记区间重置]
算法
漫随流水2 小时前
leetcode算法(515.在每个树行中找最大值)
数据结构·算法·leetcode·二叉树
mit6.8242 小时前
dfs|前后缀分解
算法
扫地的小何尚3 小时前
NVIDIA RTX PC开源AI工具升级:加速LLM和扩散模型的性能革命
人工智能·python·算法·开源·nvidia·1024程序员节
千金裘换酒4 小时前
LeetCode反转链表
算法·leetcode·链表