用于解决一些元素分组的问题,管理一系列不相交的集合,并支持两种操作:
合并(Union):把两个不相交的集合合并为一个集合。
查询(Find):查询两个元素是否在同一个集合中。
思想:用集合中的一个元素代表集合。最开始,对于只有一个元素的集合,代表元素是唯一的那个元素;通过合并操作,得到一个树状的结构,代表元素在根节点上,其它元素指向代表元素;要寻找集合的代表元素,只需要一层一层往上访问父节点,直达树的根节点。
路径压缩:尽量使集合中的每个节点的父节点都为根节点,以避免查找时并查集效率低。查找过程会把整个查找路径压缩,下次查找时可以提高效率。
按秩合并:将集合中元素个数存储在集合代表节点中,合并时将rank小的集合合并到rank大集合。
- 初始化:建立数据结构,记录每个元素的父节点,集合的秩;
- 查找:递归找当前节点的父节点,如果父节点不是代表节点,将其顺便设置为代表节点;
- 合并:比较两个元素所在集合的秩,将秩小的集合合并到秩大的集合中。
int fa[MAXN];
int rank[MAXN];
void init(int n) // 初始化
{
for (int i = 1; i <= n; ++i)
{
fa[i] = i;
rank[i] = 1; // 秩
}
}
int find(int x) // 查找并压缩路径
{
if(x == fa[x])
return x;
else{
fa[x] = find(fa[x]); //父节点设为根节点
return fa[x]; //返回父节点
}
}
void merge(int i, int j) // 合并
{
int x = find(i), y = find(j); //先找到两个根节点
if (rank[x] <= rank[y])
fa[x] = y;
else
fa[y] = x;
if (rank[x] == rank[y] && x != y)
rank[y]++; //如果深度相同且根节点不同,则新的根节点的深度+1
}