并查集由一个整数型的数组和两个函数构成。数组pre[]记录了每个点的前导点是什么,函数find是查找,join是合并。
int pre[1000 ];
int find(int x) //查找根节点
{
int r=x;
while ( pre[r ] != r ) //返回根节点 r
r=pre[r ];
int i=x , j ;
while( i != r ) //路径压缩
{
j = pre[ i ]; // 在改变上级之前用临时变量 j 记录下他的值
pre[ i ]= r ; //把上级改为根节点
i=j;
}
return r ;
}
void join(int x,int y) //判断x y是否连通,
//如果已经连通,就不用管了 //如果不连通,就把它们所在的连通分支合并起,
{
int fx=find(x),fy=find(y);
if(fx!=fy)
pre[fx ]=fy;
}
find函数是找出以自己为前导点的点(最前导点)。
find函数中的路径压缩。为了提高查找效率,最理想的情况就是所有点的前导点就是集合的最前导点,一共就两级结构,只要查找一次。哪怕不能完全做到,也最好尽量接近。每次查询都做了优化处理,所以整个树的层数都会维持在比较低的水平上。
join函数在两个互不联通的点集合中找出各自的最前导点。然后将两个连线(其中一个成为另一个的前导点),这样两个点集合就联通了。