一些灵巧的求并算法

class DisjSets
{
public:
explicit DisjSets(int numElements);
int find(int x)const;
int find(int x);
void unionSets(int root1, int root2);
private:
vector<int> s;
};

DisjSets::DisjSets(int numElements) : s(numElements)
{
for (int i = 0; i < s.size(); i++)
s[i] = -1;
}

//下面是将两个集合并
void DisjSets::unionSets(int root1, int root2)
{
s[root2] = root1;
}

//下面是寻找一个元素所在的集合
int DisjSets::find(int x) const
{
if (s[x] < 0)
return x;
else
return find(s[x]);
}

//下面的是并集的一个变种，用于控制树的高度以及节点数量
void DisjSets::unionSets(int root1, int root2)
{
if (s[root2] < s[root1])    //root2 is deeper
s[root1] = root2;
else
{
if (s[root1] == s[root2])
s[root1]--;            //如果深度相同的话，再是root1的深度变得更深一点
s[root2] = root1;        //然后再让root2指向root1
}
}

//对上述并集程序使用了路径压缩之后，体现在find程序上的变化
int DisjSets::find(int x)const
{
if (s[x] < 0)
return x;
else
return s[x] = find(s[x]);
}