disjoint_set并查集 :可用于统计森林里有几棵树
//disjoint_set并查集
#include <iostream>
using namespace std;
int parent[1000];
//初始化数组
void initalise(int parent[]) {
for(int i=0; i<=1000; i++) {
parent[i] = i; //初始的时候每个结点都是独立的结点,其根结点就是自己
}
}
//查
int find_root(int x) {
int z = x; //用z来预先记下待查找结点
while(x != parent[x]) { //没找到根结点,就顺着自身的父结点向上找,直至i=parent[i],此时这些结点有根结点i
x = parent[x];
}
//为了提高查找效率,必须做如下优化
while(z != parent[z]) {
int t = z; //暂时保存开始时候查找的结点
z = parent[z];//将z指向它的前一个父结点
parent[t] = x; //将z从自己结点走到根结点的路径上的所有结点的根结点全部改为最终的根结点,
//这样一来,下次查找只要从当前结点直接可以查找到根结点,不需要逐个结点挨个查找他们的父亲结点
}
return x;
}
//查
void union_set(int a, int b) {
int a_root = find_root(a);
int b_root = find_root(b);
parent[a_root] = b_root; //将结点a的根结点指向b的根结点即将a并入b的根结点下
}
int main(){
initalise(parent);
union_set(8, 2); //将8的根结点指向2
union_set(8, 3); //将8的1根结点2也指向3 , 此时2 3 8的根结点均为3
union_set(3, 9);
union_set(9, 10); //此时2 3 8 9 均指向根结点10
cout << find_root(2) << " " << find_root(3) << " " << find_root(8) << endl;
int sum = 0; //统计有几棵树
for(int i=1; i <= 1000; i++) {
if(i == find_root(i)) { //找到了相同的根结点即为一棵树
sum++;
}
}
cout << sum;
return 0;
}