并查集模板

最新推荐文章于 2021-12-15 00:30:14 发布

jiangheng12138

最新推荐文章于 2021-12-15 00:30:14 发布

阅读量151

点赞数

分类专栏：数据结构与算法文章标签：数据结构

本文链接：https://blog.csdn.net/automan12138/article/details/115151638

版权

并查集路径压缩根节点数据结构 C++实现

关键词由CSDN通过智能技术生成

数据结构与算法专栏收录该内容

1 篇文章 0 订阅

订阅专栏

并查集模板

//基于parent 和 size 的优化：
//初始化
int parent[n];
int size[n];
for(i = 0; i < n; i++) {
	parent[i] = i;
	size[i] = 1; //初始化时每一个节点作为根，子节点 只有他自己
}
//查找a元素的根节点
int find(int a) {
	assert a >= 0 && a < n
	
	int parenta = parent[a];
	while(parenta != parent[parenta]) {
		parent[parenta] = parent[parent[parenta]];   //路径压缩
		parenta = parent[parenta];
	}
	return parenta;
}
// 判断两元素是否连接
BOOL isConnected(int a, int b) {
	return (find(a) == find(b));
}
// 拼接
void union(int p, int q) {
	int p_parent = find(p);
	int q_parent = find(q);

	if (p_parent != q_parent) {
		if (p_parent < q_parent) {
			parent[p_parent] = q_parent; 
		} else if (p_parent > q_parent) {
			parent[q_parent] = p_parent;
		} else {
			parent[q_parent] = p_parent;
			size[p_parent]++;
		}
	}
}