集合 | 并查集（树结构表示，数组存储）| C语言

一、复杂表示

• 查找某个元素所在集合（用根结点表示）

int Find(SetType S[], ElementType X)
{   // 在数组S中查找值为X的元素所属的集合
    // MaxSize是全局变量，为数组S的最大长度
    int i;
    for (i = 0; i < MaxSize && S[i].Data != X; i++);
    if (i >= MaxSize) return -1;   // 未找到X，返回-1
    for (; S[i].Parent >= 0; i = S[i].Parent);
    return i;   // 找到X所属集合，返回树根结点在数组S中的下标
}

• 集合的并运算

void Union(SetType S[], ElementType X1, ElementType X2)
{   // 将两个元素分别所在的集合合并
    int Root1, Root2;
    Root1 = Find(S, X1);  // X1的根结点的数组下标
    Root2 = Find(S, X2);  // X2的根结点的数组下标
    if (Root1 != Root2) S[Root2].Parent = Root1; // 将X1根结点的s数组下标赋给X2根结点的Parent
}

缺点是表示并集的树高会增加得很快，因此得用按秩归并

二、简化表示

• 数组下标表示元素，数组成员表示父结点，负数表示根结点

#define MAXN 1000  // 集合最大元素个数
typedef int ElementType;  // 默认元素可以用非负整数表示
typedef int SetName;      // 默认用根结点的下标作为集合名称
typedef ElementType SetType[MAXN];  // 假设集合元素下标从0开始

void Union(SetType S, SetName Root1, SetName Root2)
{   // 这里默认Root1和Root2是不同集合的根结点
	// 按秩归并 -- 比规模（根结点的数组成员大小表示集合的规模）
    // 保证小集合并入大集合
    if (S[Root2] < S[Root1]) {  // 如果集合2比较大
        S[Root2] += S[Root1];   // 集合1并入集合2
        S[Root1] = Root2;
    }
    else {                      // 如果集合1比较大
        S[Root1] += S[Root2];   // 集合2并入集合1
        S[Root2] = Root1;
    }
}

路径压缩

SetName Find(SetType S, ElementType X)
{   // 默认集合元素全部初始化为-1
    if (S[X] < 0) // 找到集合的根
        return X;
    else
    	// 路径压缩，先找到根，把根变成X的父结点，再返回根
        return S[X] = Find(S, S[X]); 
}

尾递归容易使系统堆栈爆掉，也容易转换成循环，因此编译器会将尾递归转化成循环，进行优化