并查集
一种非常精巧且实用的数据结构,用来处理不相交集合的合并、查找。
并查集的基本操作
1. 初始化
定义int s[i],
(s为set的缩写,意思是集合)
初始化,令s[i]=i。
void init_set(){
for(int i=1;i<=N;i++)
s[i]=i;
}
2. 合并
例如,加入第一个朋友关系(1,2)
- 在并查集s中,把节点1合并到节点2,
- 也就是把节点1所属于的的集1,改成节点2所属的集2
加入第二个朋友关系(1,3)
- 查找结点1属于的集,是节点2的集,递归查找节点2的集是集2
- 把节点2的集:集2合并到节点3的集:集3。
- 此时,节点1,2,3都属于一个集:节点三的集:集3
加入第三个朋友关系(2,4)
void union_set(int x,int y){
x=find_set(x);
y=find_set(y);
if(x!=y)s[x]=s[y];
}
3. 查找
查找元素的集,是一个递归,直到节点元素的值和它的集相等,就找到了根节点的集。
int find_set(int x){
return x==s[x]?x:find_set(s[x]);//三元运算符,若x=s[x]返回x,否则返回递归。
}
基本并查集的时间复杂度
查找find_set()、合并union_set()的搜索深度是树的长度,复杂度为O(n)
并查集的优化
合并的优化(一般不用,而用路径压缩)
合并两个节点时,先搜索到他们的根节点:
合并这两个根节点:把一个根节点的及改成另一个根节点。
这两个根节点的高度不同,把高度较小的集合合并到较大的集上,能减少树的高度。
! 路径压缩(查询优化) !
!!!要用并查集,必用路径压缩!!!
查询find_set():沿着搜索路径找到根节点,这条路径可能很长。
优化:沿路径返回时,顺便把i所属的集改为根节点.
这样,当下次在搜索时,复杂度就为O(1)
- 无路径压缩:
int find_set(int x){
return x==s[x]?x:find_set(s[x]);//三元运算符,若x=s[x]返回x,否则返回递归。
}
- 有路径压缩:
int find_set(int x){
if(x!=s[x])s[x]=find_set(s[x]);//路径压缩
return s[x];
- 自最深find_set(s[x])值为这个集中最深得"帮主":即4.
- 在第一次返回时s[x]=find_set(s[x]) 将节点3 所属的集改为集4,
- 在第二次返回时s[x]=find_set(s[x]) 将节点2 所属的集改为3所属的集即:集4
- 在最后一次返回时s[x]=find_set(s[x]) 将节点1 所属的集改为2所属的集即:集4
从而得到右边高度为2的树