并查集的相关操作

并查集的相关操作

#include <iostream>
#define SIZE 100
int UFSize[SIZE];//集合元素数组(双亲指针数组)

//并查集初始化
void Initial(int S[]) {
    for (int i = 0; i < SIZE; i++)//每个自成单元素集合
        S[i] = -1;
}

//Find,"查"操作，找x所属集合(返回x所属根结点)
int Find1(int S[],int x) {
    while (S[x] >= 0)//循环寻找x的根
        x = S[x];
    return x;//根的S[]小于0
}

//Union,"并'操作，求两个不相交子集合的并集
void Union1(int S[], int Root1, int Root2) {
    if (Root1 == Root2)//要求Root1与Root2是不同的集合
        return;
    S[Root2] = Root1;//将根Root2连接到另一根Root1下面
}

//Union,"并'操作，小树合并到大树
void Union2(int S[], int Root1, int Root2) {//此时根结点在数组中的值为该树的结点总数
    if (Root1 == Root2)//要求Root1与Root2是不同的集合
        return;
    if (S[Root2] < S[Root1])//Root2结点数更少
    {
        S[Root1] += S[Root2];//累加结点总数
        S[Root2] = Root1;//小树合并到大树
    }
    else
    {
        S[Root2] += S[Root1];//累加结点总数
        S[Root1] = Root2;//小树合并到大树
    }
}

//Find操作的优化，先找到根结点，再进行"压缩路径"
int Find2(int S[], int x) {
    int root = x;
    while (S[root] >= 0)//循环寻找根
        x = S[x];
    while (x != root) {//压缩路径
        int t = S[x];//t指向x的父结点
        S[x] = root;//x直接挂到根结点下
        x = t;
    }
    return root;//返回根结点编号
}