#include
#include
using namespace std;
class UF{
public:
UF(int size): vec(size), parent(size) {
for (int i = 0; i < size; i++) {
vec[i] = i;
parent[i] = -1;
}
}
void Union(int p, int q) {
int root1 = Find(p);
int root2 = Find(q);
//将小树链接到大树,大小则由数的结点个数确定
if (root1 > root2) { // 例如-2 > -3 说明要将root1链接到root2上面
parent[root2] += parent[root1]; //更新大树的结点个数
parent[root1] = root2; //将树1链接到树2
} else {
parent[root1] += parent[root2];
parent[root2] = root1;
}
}
int Find(int x) {
if (parent[x] < 0) //递归直到x的父节点为负数
return x;
else
return parent[x] = Find(parent[x]); //带有路径压缩功能,若本次找到组号,那么递归找到组号后直接赋值给parent[x],下次查找则能在O(1)内完成
}
int count(int x) {
int c = parent[Find(x)];
return -c;
}
private:
vector vec;
vector parent; //parent有两个作用,一个是指示集合中的个数,另外若parent[x]为负数,x则为集合
//中的组号
};
int main() {
UF uf(10);
uf.Union(2, 3);
uf.Union(6, 7);
uf.Union(0, 1);
uf.Union(1, 2);
uf.Union(3, 6);
cout << uf.Find(0) << endl;
cout << uf.Find(1) << endl;
cout << uf.Find(2) << endl;
cout << uf.Find(3) << endl;
cout << uf.Find(6) << endl;
cout << uf.Find(7) << endl;
cout << uf.count(0) << endl;
cout << uf.count(1) << endl;
cout << uf.count(2) << endl;
cout << uf.count(3) << endl;
cout << uf.count(6) << endl;
cout << uf.count(7) << endl;
return 0;
}
未完待续