挂个例题:落谷亲戚
并查集是最简洁而优雅的算法之一,主要用于解决一些元素分组的问题。管理一系列不相交的集合,支持两种操作:
- 合并 :把领个不相交的集合合并为一个集合 。
- 查询 :查询两个元素是否在同一个集合中。
并查集的重要思想在于,用集合中的一个元素代表集合。有个大佬给出过有趣比喻,把集合比喻成帮派,而代表元素则是帮主。
这样就能写出最基础的代码:
初始化:
假设每个人各为一派掌门。数组a用来春初父节点。
inline void init(int n)
{
for(int i = 0 ; i <n ; i++)
a[i] = i ;
}
查询:
用递归的额写法实现对代表元素的查询:一层一层访问父节点,直至根节点(根节点的标志就是父节点是本身)。要判断两个元素是否属于同一个集合,只需要看它们的根节点是否相同即可。
int find(int x)
{
if(a[x] == x)
return x ;
else
return find(a[x]) ;
}
合并:
合并操作也是很简单的,先找到两个集合的代表元素,然后将前者的父节点设为后者即可。当然也可以将后者的父节点设为前者。
inline void merge(int i, int j)
{
a[find(i)] = find(j);
}