最近在学习并查集,讲真,这不就是离散数学中的等价类吗?哈哈。
它主要有查询和归并操作。提供代码如下:
#include<iostream>
int FA[1000]; //利用该数组来存储每个元素的父节点
void init(int n);//初始化并查集
int Find(int x); //查询代表元素,算法,代表元素的父节点是他本身,可以根据此判断他们的两个节点是否相
void Merge(int x, int y);//合并两个元素,只需查询其父子节点并设置好新的父子节点即可。
int Find1(int x);//直接将每个节点与根节点相连
using namespace std;
int main() {
int n;
cout << "请输入并查集中的数:" << endl;
cin >> n;
init(n); //将每个元素的父节点初始化为自己。
int x;//声明节点
cout << "请输入待查询的节点:" << endl;
cin >> x;
Find(x);
int res = FA[x];
cout << "节点" << x << "的根节点为" << res << endl;
for (int i = 0; i < n; i++) {
if (i) {
Merge(i, 2 * i + 1);
}
if (i % 2 == 0) {
Merge(i + 1, i * 2);
}
}
cout << "节点" << n - 2 << "的根节点为:" << Find(n - 2) << endl;
system("pause");
return 0;
}
void init(int n) { //初始化节点的父节点
for (int i = 0; i < n; i++) {
FA[i] = i;
}
}
int Find(int x) { //递归终止条件是它自己等于自己,不终止条件是它的往上,所以可以递归走起来。
if (FA[x] == x) {
return x;//返回该元素的父节点
}
else {
Find(FA[x]);//递归查询父节点
}
}
void Merge(int x, int y) { //归并两个集合
FA[Find(x)] = Find(y);//将x所在集合的节点设置为y所在结合的父节点即可
}
int Find1(int x) { //寻找节点x的根节点
if (x = Find1(x)) {
return x;
}
else {
FA[x] = Find1(FA[x]); //路径压缩优化可以降低很多时间复杂度了
return FA[x];
}
}
运行结果如下: