记录一下算法分析课要求的作业:
用C++实现并查集算法,要求实现路径压缩+权重合并两个功能
首先,我认为弄懂具体的概念很重要(比如我看概念就看了很久)
网上有许多介绍,但大部分是只讲了并查集。
在一定的了解之后,再着手思考各种函数的思路,应该如何写,在此附上我认为讲的比较清楚的链接并查集的总结
简单介绍一下:
加权合并规则是把节点较少的一棵树与节点较多的一棵树归并时,把节点较少树的根节点指向节点较多树的根节点,这样可以把树的整体深度限制在nlogn。当两棵待归并的树的规模同样大时,使结点值较大的根结点作为值较小的根结点的子结点。
路径压缩是一种可以产生极浅的树的方法,当查找一个节点X的根节点时,可以采用路径压缩方法,设置根节点为R,则路径压缩把由X到R的路径上的每个节点的父指针都设置为直接指向R。首先要找到R,然后顺着由X到R的路径把每个节点的父指针域都设置指为R。
代码如下:
#include<iostream>
using namespace std;
#define N 16 //假定存储有16个结点,从下标0开始
//即结点序号为0-15,可自行修改
int p[N]; //存储每个结点指向的父指针
int rk[N]; //深度
void init() { //初始化
for (int i = 0; i < N; i++) {
p[i] = i; //初始化父结点都为自己本身
rk[i] = 1; //初始化深度都为1
}
}
int find(int t) { //查询树的根,实现路径压缩
if (p[t] == t) //当此结点为父结点时,直接返回
return t;
p[t] = find(p[t]); //将所有走过的却不是祖先节点的父亲都改成祖先节点!
return p[t];
}
bool differ(int a, int b) { //判断两个结点是否在不同的树里
return find(p[a]) != find(p[b]); //true则代表不相同
}
void Union(int a, int b) { //合并
if (!differ(a, b)) return; //此时两个结点在同一棵树
int r1 = find(a);
int r2 = find(b);
if (rk[r1] < rk[r2]) {
p[r1] = r2;
}
else {
p[r2] = r1;
if(rk[r1]==rk[r2])
rk[r1]++;
}
}
void Display() { //展示
for (int i = 0; i < N; i++) {
cout << i << " " << p[i];
cout << endl;
}
cout << endl;
}
int main() {
cout << "将树初始化,每个结点的父结点指向自己\n";
init();
Display();
cout << "Union: (0,2),(1,2),(3,4),(3,1),(3,5),(9,11),(12,14),(3,9),(4,14),(6,7),(8,10),(8,7),(7,0),(10,15),(10,13)\n";
Union(0, 2);
Union(1, 2);
Union(3, 4);
Union(3, 1);
Union(3, 5);
Union(9, 11);
Union(12, 14);
Union(3, 9);
Union(4, 14);
Union(6, 7);
Union(8, 10);
Union(8, 7);
Union(7, 0);
Union(10, 15);
Union(10, 13);
cout << "show the result:\n";
Display();
return 0;
}
p.s.记录一下遇到的问题:
1.不可将数组rk命名为rank、size等名字,会报错显示”rank不明确“。后来发现是因为这类名字已在C++里面被命名,下次注意。
2.本来是想实现用树的父指针表示的树来完成这次作业,后来因为种种原因(我太懒了) 放弃了,有兴趣的可以参考这个链接树的父指针表示与Union/Find算法实现