1. 定义
并查集是一种树型的数据结构,用于处理一些不相交集合的合并及查询问题。
并:即union,合并两个集合;
查:即find,判断两个集合是否在同一集合;
集:即set,集合。
并查集用一个数组father[n]实现,father[i]表示元素i的父亲结点。
2. 基本操作
2.1 初始化
最初,每个元素都是一个独立的集合,所以把每个元素的父结点都初始化为元素本身。
for(int i=1;i<=n;i++)
father[i] = i;
2.2 查找
查找给定结点的根结点。一般使用递归或递推的方式查找根结点,即反复寻找结点的父结点,直到找到father[i]==i为止。
//使用递归方式查找根结点
int findFather(int i)
{
if(father[i] == i)
return i;
else
return findFather(father[i]);
}
//使用递推方式查找根结点
int findFather(int i)
{
while(father[i] != i)
i = father[i];
return i;
}
当元素数量很多时,使用以上查找方式效率较低,为了优化代码,可以把当前查询结点的路径上的所有结点的父亲都指向根结点。
//使用递归方式查找根结点
int findFather(int i)
{
if(i == father[i])
return i;
else
{
int f = findFather(father[i]);
father[i] = f;
return f;
}
}
//使用递推方式查找根结点
int findFather(int i)
{
int x = i;
while(i != father[i])
i = father[i];
while(x != father[x])
{
int y = x;
x = father[x];
father[y] = i;
}
return i;
}
2.3 合并
判断两个元素是否属于同一集合(即根结点是否相同),只有两个元素不属于同一个集合时才需要合并,合并过程就是将其中一个集合的根结点改为另一个集合的根结点(只改根结点)。
void union(int a,int b)
{
int fatherA = findFather(a);
int fatherB = findFather(b);
if(fatherA != fatherB)
father[fatherB] = fatherA; //注意是将B集合的根结点的父结点改为A集合的根结点
}