1 class DisjSets
2 {
3 public:
4 explicit DisjSets(int numElements);
5 int find(int x)const;
6 int find(int x);
7 void unionSets(int root1, int root2);
8 private:
9 vector<int> s;
10 };
11
12
13 DisjSets::DisjSets(int numElements) : s(numElements)
14 {
15 for (int i = 0; i < s.size(); i++)
16 s[i] = -1;
17 }
18
19 //下面是将两个集合并
20 void DisjSets::unionSets(int root1, int root2)
21 {
22 s[root2] = root1;
23 }
24
25 //下面是寻找一个元素所在的集合
26 int DisjSets::find(int x) const
27 {
28 if (s[x] < 0)
29 return x;
30 else
31 return find(s[x]);
32 }
1 //下面的是并集的一个变种,用于控制树的高度以及节点数量
2 void DisjSets::unionSets(int root1, int root2)
3 {
4 if (s[root2] < s[root1]) //root2 is deeper
5 s[root1] = root2;
6 else
7 {
8 if (s[root1] == s[root2])
9 s[root1]--; //如果深度相同的话,再是root1的深度变得更深一点
10 s[root2] = root1; //然后再让root2指向root1
11 }
12 }
1 //对上述并集程序使用了路径压缩之后,体现在find程序上的变化
2 int DisjSets::find(int x)const
3 {
4 if (s[x] < 0)
5 return x;
6 else
7 return s[x] = find(s[x]);
8 }