C++ 树的应用（二）：集合及其运算

本文为浙大版《数据结构》学习笔记。
可用树的结构表示集合，树的每个结点表示一个集合元素。

一、集合的表示

定义一个结构数组，其中一个变量存储集合中元素的数值，另一个变量存储元素的父结点（即所属集合）。若为根结点，令其父结点为负值（不存在）。
集合的类型描述：

#define MaxSize 15
typedef int ElementType;
struct SetType{
    ElementType Data;
    int Parent;
};

（一）查找

/*查找元素X所属集合的根结点在数组中的下标*/
int Find(SetType s[],ElementType X)
{
    int i;
    for(i=0;i<MaxSize && s[i].Data!=X;i++);
    if(i>=MaxSize)
        return -1;
    for(;s[i].Parent>=0;i=s[i].Parent);
    return(i);
}

（二）合并

void Union(SetType s[],ElementType X1,ElementType X2)
{
    int Root1,Root2;
    Root1=Find(s,X1);
    Root2=Find(s,X2);
    if(s[Root1].Parent<s[Root2].Parent){
        s[Root1].Parent+=s[Root2].Parent;
        s[Root2].Parent=Root1;
    }
    else{
        s[Root2].Parent+=s[Root1].Parent;
        s[Root1].Parent=Root2;
    }
}

二、集合的简化表示

将集合中的N个元素映射为整数0~N-1，可直接用数组的下标表示集合中的元素，数组中存储元素的父结点所在的位置。
集合的类型描述：

#define MAXN 10
typedef int ElementType;
typedef int SetName;
typedef ElementType SetType[MAXN];

（一）查找
查找编号为X的元素所属集合（返回根结点在数组中的下标）
1.普通查找
沿着该元素的父结点一级一级往上查，当发现一个元素的指针值域为负数，该元素就是X所属集合的树根结点。

/*寻找X所在的集合（用根结点在数组中的下标表示*/
SetName Find1(SetType S,ElementType X)
{
    for(;S[X]>=0;X=S[X]);
    return(X);
}

2.路径压缩
在第一次查找时，使得所有被遍历元素直接变为根结点的孩子，这样，当下次再查找时，就不用再一级一级比较，只需比较两次就够了。可用递归实现。

/*(路径压缩)寻找X所在的集合（用根结点在数组中的下标表示*/
SetName Find(SetType S,ElementType X)
{
    if(S[X]<0)
        return X;
    else
        return(S[X]=Find(S,S[X]));
}

（二）并运算
将元素X1和X2所属的两个集合进行合并，找到两个元素所在集合树的根结点，如果不同根，则将其中一个根结点的父结点值域设置成另一个根结点的数组下标。
1.随意合并
不管集合的大小，直接将一个集合合并入另一个集合。

/*随意合并，效率不高*/
void Union1(SetType S,SetName Root1,SetName Root2)
{
    S[Root2]=Root1;
}

2.按秩合并
将规模（高度）较小的集合合并到规模（高度）较大的集合中，这样可减缓集合规模的增长，提高遍历的效率。

/*按秩合并*/
void Union(SetType S,SetName Root1,SetName Root2)
{
    if(S[Root2]<S[Root1]){
        S[Root2]+=S[Root1];
        S[Root1]=Root2;
    }
    else{
        S[Root1]+=S[Root2];
        S[Root2]=Root1;
    }
}

可直接运行的程序：

#include <iostream>
#define MAXN 10
typedef int ElementType;
typedef int SetName;
typedef ElementType SetType[MAXN];
using namespace std;

int main()
{
    SetName Find1(SetType S,ElementType X);
    SetName Find(SetType S,ElementType X);
    void Union1(SetType S,SetName Root1,SetName Root2);
    void Union(SetType S,SetName Root1,SetName Root2);
    SetType S={6,-1,1,-1,1,3,-1,1,3,6};
    Union(S,1,3);
    cout<<Find(S,5)<<endl;
    return 0;
}

/*寻找X所在的集合（用根结点在数组中的下标表示*/
SetName Find1(SetType S,ElementType X)
{
    for(;S[X]>=0;X=S[X]);
    return(X);
}

/*(路径压缩)寻找X所在的集合（用根结点在数组中的下标表示*/
SetName Find(SetType S,ElementType X)
{
    if(S[X]<0)
        return X;
    else
        return(S[X]=Find(S,S[X]));
}

/*随意合并，效率不高*/
void Union1(SetType S,SetName Root1,SetName Root2)
{
    S[Root2]=Root1;
}

/*按秩合并*/
void Union(SetType S,SetName Root1,SetName Root2)
{
    if(S[Root2]<S[Root1]){
        S[Root2]+=S[Root1];
        S[Root1]=Root2;
    }
    else{
        S[Root1]+=S[Root2];
        S[Root2]=Root1;
    }
}

（三）具体应用

判断联通问题（两个元素是否属于同一个集合）
题目描述：

代码：

#include <iostream>
#define MAXN 10
typedef int ElementType;
typedef int SetName;
typedef ElementType SetType[MAXN];
using namespace std;

int main()
{
    void Initialization(SetType S,int n);
    void Input_connection(SetType S);
    void Check_connection(SetType S);
    void Check_network(SetType S,int n);

    SetType S;
    int n;
    char in;
    cin>>n;
    Initialization(S,n);
    do{
        cin>>in;
        switch(in){
            case 'I':Input_connection(S);break;
            case 'C':Check_connection(S);break;
            case 'S':Check_network(S,n);break;
        }
    }while(in!='S');
    return 0;
}

/*(路径压缩)寻找X所在的集合（用根结点在数组中的下标表示*/
SetName Find(SetType S,ElementType X)
{
    if(S[X]<0)
        return X;
    else
        return(S[X]=Find(S,S[X]));
}

/*按秩合并*/
void Union(SetType S,SetName Root1,SetName Root2)
{
    if(S[Root2]<S[Root1]){
        S[Root2]+=S[Root1];
        S[Root1]=Root2;
    }
    else{
        S[Root1]+=S[Root2];
        S[Root2]=Root1;
    }
}

/*初始化集合*/
void Initialization(SetType S,int n)
{
    int i;
    for(i=0;i<n;i++)
        S[i]=-1;
}

/*将两个元素并入同一集合(联通路径)*/
void Input_connection(SetType S)
{
    ElementType u,v;
    SetName Root1,Root2;
    cin>>u>>v;
    Root1=Find(S,u-1);
    Root2=Find(S,v-1);
    if(Root1!=Root2)
        Union(S,Root1,Root2);
}

/*判断两个元素是否属于同一集合*/
void Check_connection(SetType S)
{
    ElementType u,v;
    SetName Root1,Root2;
    cin>>u>>v;
    Root1=Find(S,u-1);
    Root2=Find(S,v-1);
    if(Root1==Root2)
        cout<<"yes"<<endl;
    else
        cout<<"no"<<endl;
}

/*查看元素是否完全联通*/
void Check_network(SetType S,int n)
{
    int i,counter=0;
    for(i=0;i<n;i++)
        if(S[i]<0) counter++;
    if(counter==1)
        cout<<"the network is connected."<<endl;
    else
        cout<<"the number of componebts is:"<<counter<<endl;
}