并查集
首先,什么是并查集:
并查集,在一些有N个元素的集合应用问题中,我们通常是在开始时让每个元素构成一个单元素的集合,然后按一定顺序将属于同一组的元素所在的集合合并,其间要反复查找一个元素在哪个集合中。这一类问题近几年来反复出现在信息学的国际国内赛题中。其特点是看似并不复杂,但数据量极大,若用正常的数据结构来描述的话,往往在空间上过大,计算机无法承受;即使在空间上勉强通过,运行的时间复杂度也极高,根本就不可能在比赛规定的运行时间(1~3秒)内计算出试题需要的结果,只能用并查集来描述。
并查集是一种树型的数据结构,用于处理一些不相交集合(disjoint sets)的合并及查询问题。常常在使用中以森林来表示。
声明:以上内容出自并查集_百度百科 (baidu.com)。
并查集作为一种记录森林的数据结构,每个数据存储所在数的根节点判断两个数是否在同一个树只用查看更节点是否相同
。
实现
第一步,预处理
a i a_i ai 原本的根都是自己,所有要预处理。这块一定要有,我之前被坑惨了,代码:
for (int i = 1; i <= n; i ++) {
a[i] = i;
}
第二步, find \text{find} find 函数。
用作更新,采用递归的思路,分为路径压缩和非路径压缩两种。
先来看为路径压缩的方法,如果 a x = x a_x = x ax=x,说明找到了,返回 x x x,代码:
int find(int x) {
return (a[x] == x ? x : find(a[x]));
}
路径压缩也就是在更新原本数据的同时,把路上所有的数据都更新了,代码:
int find(int x) {
return (a[x] == x ? x : a[x] = find(a[x]));
}
第三步,合并+判断
合并:我们可以把 x x x 的根节点合并到 y y y 的根节点,把 x x x 所在的树和 y y y 所在的树合并为一个树,这里就不放代码了。
判断:我们在之前讲过了,要判断两个节点是否在用一棵树里面,只需要判断根节点是否相同,所以我们只需判断 find ( x ) \text{find}(x) find(x) 是否等于 find ( y ) \text{find}(y) find(y),就可以判断是否在同一棵树里面了。
综上所述,我们就可以写出一个 AC 的
并查集代码了。
例题
直接使用上面的方法,AC。
代码:
练习
我在这里推荐一些题目:
资料
此文章同步发表在https://ztrztr.oier.space/post/bset和https://www.luogu.com.cn/blog/zzttrr/bing-zha-ji