主要函数:
int find(int x){ int r=x; while(r!=pre[r]) r=pre[r]; } void join(int x,int y){ int fx=find(x),fy=find(y); if(fx!=fy) pre[fx]=fy;//把fx当做fy的子节点 else { if(fx==fy) //相等的话可以判断有环 } }
路径压缩之递归优化:(使用递归优化可以遍历到每一个节点,在进行回溯的时候我们可以对子节点与父节点的关系进行处理,在解决带权并查集时常用到)
但是当数据量过大时,就不能用了,会MLE错误
int find(int x){ if(x==pre[x]) return x; else { int k=pre[x]; pre[x]=find(pre[x]); return fa[x]; } }
路径压缩非递归
int find(int x){ int r=x; while(pre[r]!=r){ r=pre[r]; } int k=x,j; while(k!=r){ j=pre[k];//用j保存当前节点的父节点 pre[k]=r;//修改父节点 k=j; } return k; }