第21章：用于不相交集合的数据结构

一：不相交集类的声明如下：

#ifndef DISJSETS_H
#define DISJSETS_H
#include <vector>
using namespace std;
class DisjSets{
public:
        explicit DisjSets(int);

        bool isSameSet(int x, int y) const
        { return findSet(x)==findSet(y);}

        int findSet(int) const;
        int findSet(int);

        void unionSets(int,int);
private:
        vector<int> s;
};
#endif

二：不相交集类的实现如下：

#include "DisjSets.h"
DisjSets::DisjSets(int numElements):s(numElements)
{
        for(int i=0;i!=s.size();i++)
                s[i]=-1;
}

int DisjSets::findSet(int x) const
{
        if(s[x]<0)
                return x;
        return findSet(s[x]);
}

int DisjSets::findSet(int x)
{
        if(s[x]<0)
                return x;
        return s[x]=findSet(s[x]);
}

void DisjSets::unionSets(int x,int y)
{
        int root1=findSet(x);
        int root2=findSet(y);

        if(root1==root2)
                return;

        if(s[root2]<s[root1])
                s[root1]=root2;
        else{
                if(s[root2]==s[root1])
                        s[root1]--;
                s[root2]=root1;
        }
}