并查集是用来处理一些相交的集合的合并问题,使用并查集可以讲一些有关系的集合用一个节点表示出来,一般操作有下面几个:
1、初始化
for (int i = 1; i <= n; i++)//n表示输入的结点的个数
{
pre[i] = i;//将每一个结点的前导点设置为自己
}
2、find()函数
find()函数一般有两种写法,递归和非递归
int find(int x)//递归
{
if (x != pre[x])
pre[x] = find(pre[x]);//如果父节点不等于自身,寻找他的父节点并标记
return pre[x];
}
int find(int x)//非递归
{
int r = x;
while (pre[r] != r)
r = pre[r];
int i = x, j;
while (i != r)//路径压缩算法
{
j = pre[i];//记录x的前导结点
pre[i] = r;//将i的前导结点设置为r根节点
i = j;
}
return r;
}
其中第二种写法还可以处理带权值的并查集
3、合并操作join()
这个就很好理解了,就是把两个父节点相同的集合合并起来
void join(int x, int y)
{
int a = find(x);
int b = find(y);
if (a != b)//如果a,b不是相同的根节点,则说明a和b不连
{
pre[a] = b;//我们将a和b相连 将a的前导结点设置为b
}
}
并查集的模板题HDU1232 畅通工程 http://acm.hdu.edu.cn/showproblem.php?pid=1232
AC代码 https://blog.csdn.net/loven0326/article/details/81487937
带权值的并查集就是加一个val[]数组来记录权值,用递归的find()函数来实现
int find(int x)
{
if (x == pre[x]) return x;
int t = pre[x];
pre[x] = find(pre[x]);
val[x]+= val[t];//记录权值
}