并查集
并查集:
** 作用**:用于快速判断元素是否在一个集合中,或者合并两个集合的数据结构;
原理:使用树结构来存储元素:每个节点x的父节点为p[x],所以树根节点x的父节点还是x;
寻找根节点和路径压缩示意
1.判断树根:`if(p[x]==x) //树根的父节点就是自己.
2.从某个节点开始寻找树根:while(p[x] != x) x=p[x];
3.判断两个元素是否在同一个集合中,只需要比较这两个元素的根节点是否一样即可。
4.合并集合,将一个集合的树根的父节点指向另一个集合的树根。。比如集合a,b,先将a合并到b里面,具体操作为 p[a]=b;
5.优化(路径压缩):当从一个节点开始寻找父节点,会把路径上的所有节点的父节点都指向树根,
代码
路径压缩寻找根节点(递归写法):
int find(int x){
if(p[x] != x) p[x]=find(p[x]);//寻找根节点,最后返回的值是根节点,回溯过程将每个节点的父节点全部赋值位根节点。
return p[x];
}
非递归写法/防止爆栈/(写找出根节点,然后把每个节点的父节点全部赋值为根节点,)
```cpp
int find(int x){
int y=x,z;//变量z用于路径压缩;
while(p[y]!=y) y=p[y];//寻找根节点;
while(x != y){
z=p[x];//先将x的父节点存储起来,
p[x]=y;//将x的父节点指向根节点;
x=z;//给x赋值为x的父节点。
}
return r;
连通块问题(使用size数组维护)
size数组用于存储一个集合(根节点)中的元素。
比如将两个集合a,b合并
if(find(a)==find(b)) continue;//必须要进行特判,假如两个集合已经是同一个集合,则size不需要改变。
else{ Size[find(b)]+=Size[find(a)];
p[find(a)]=find(b);