数据结构-树并查集的定义及其操作

Posted 2022-12-05 Mount256

文章目录

• • 1 数据结构定义
  • 2 初始化
  • 3 查找操作
  • 4 并操作

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