作用
查:给定一个元素,查询它在哪个集合内
并:合并两个元素所在的集合
实现思路
对应关系
元素-->结点
集合-->树
多个集合-->森林
用树的根节点作为不同树的标志
合并时只需要将根节点链接
实现
用数组表示树,数组下标表示元素值,数组的值表示该元素对应的父亲结点
father[i] = j : 元素i的父亲结点是j
对于根节点 father[i] = i
图中有两个集合,由两个树表示,根节点分别为元素2,3
查找元素4:
合并元素4,0所在集合:
代码
#include<stdio.h>
const int maxn = 10;
int father[maxn];
void init(int n){ //初始化,开始时,所有元素都不在同一个集合,自身构成集合
for(int i = 0; i < n; i++)
father[i] = i;
}
int find(int x){ //查找元素x所在的集合的根节点
while(x != father[x])
x = father[x];
return x;
}
void Union(int x1, int x2){ //合并x1和x2所在的集合
int Fax1, Fax2; //分别寻找x1,x2的根节点
Fax1 = find(x1);
Fax2 = find(x2);
if(Fax1 != Fax2) //如果x1,x2不在同一个集合,合并
father[Fax1] = Fax2;
}
改进:路径压缩
有时路径过长会导致find函数内部的while循环要执行很多次才能找到根节点
查找0时需要执行多次while循环
对find函数进行改进
int find(int x){
int a = x;
while(x != father[x])
x = father[x];
//路径压缩
while(a != father[a]){
father[a] = x;
a = father[a];
}
return x;
}
压缩后: