说得简单点,并查集就是一个元素的集合,可以迅速地查询两个元素是否在一个集合中。
并查集以森林的形式存储,要查询两个元素是否在一个集合中,首先要知道两个元素分别在哪个集合。
由于一个元素不可能同时是森林中两棵不同的树的根节点,只要查询集合的元素所在树的根节点就可以确定它在哪个集合。
查询元素的根节点的代码很好写:
int find(int x)
{
if (fa[x]==x) return x;//如果元素x没有父亲,则已经是根节点,返回元素x(这里要预处理fa[x]=x)
else return find(fa[x]);//如果元素x有父亲,则递归查询x的父亲是否是根节点
}
解决了这个问题,就可以确定两个元素是否在一个集合了:
bool cheak(int x,int y)
{
if (find(x)==find(y)) return true;//如果两个元素的根节点相同,则在同一集合
return false;//否则不在同一集合
}
哎呀好像还没说合并呢:
使节点x的根节点的父节点为节点y,则节点x所在的集合的所有节点的根节点都指向节点y的根节点,然后就完成了合并的操作
void join(int x,int y)
{
if (!cheak())//如果两个元素不在同一集合
fa[find(x)]=find(y);//使x的根节点的父节点为y的根节点
}
查找根节点的路径压缩优化:
由于合并时有点随便,导致了集合的形状可能是很奇怪,查询起来很慢,这个时候就要用到路径压缩:
在查询节点x的父节点时,同时将find(fa[x])的结果赋值fa[x],由于find(fa[x])的值就是x所在集合的根节点,直接使fa[x]指向根节点就可以压缩查询的时间复杂度。
代码就不贴出了,应该是会写的。