1 数据结构定义
#define MAX 50
int UFSets[MAX]; // 并查集
2 初始化
// 参数:并查集 S
void Init (int S[]){
int i;
for (i = 0; i < MAX; i++)
S[i] = -1;
}
【注】根结点可用来保存该子集合的元素个数(负数表示)。
3 查找操作
- 寻找包含 x 的树根:
// 参数:并查集 S,索引/下标 x
int Find (int S[], int x){
while (S[x] >= 0)
x = S[x];
return x;
}
- 压缩路径:先找到根结点,再将查找路径上所有结点都挂在根结点上
// 参数:并查集 S,索引/下标 x
int Find (int S[], int x){
int root = x;
// 寻根
while (S[root] >= 0)
root = S[root];
// 压缩路径
while (x != root){
int t = S[x]; // t 暂时保存 x 的双亲结点(不一定是根结点!)
S[x] = root; // x 直接挂在根结点下
x = t; // x 更新为双亲结点
}
return root;
}
- 压缩路径:一边查找一边压缩路径,将查找路径上所有结点都挂在根结点上(递归写法)
// 参数:并查集 S,索引/下标 x
int Find (int S[], int x){
if (S[x] < 0) // 如果就是根结点
return x;
else{ // 如果不是根结点,一边查找一边压缩路径
S[x] = Find(S[x]); // 找到其根结点,将双亲结点更新为其根结点
return S[x];
}
}
4 并操作
- 把集合 S 中的子集合 Root2 并入子集合 Root1:
void Union (int S[], int Root1, int Root2){
if (Root2 != Root1) // Root2 和 Root1 必须是不同的两个集合
S[Root2] = Root1;
}
- 把集合 S 中,包含元素 y 的子集合 Root2 并入包含元素 x 的子集合 Root1:
void Union (int S[], int x, int y){
S[Find(y)] = Find(x);
}