算法补完计划(二) 带权并查集

普通并查集

先令每个fa[i] = i

路径压缩操作

int find(int x){
	if(fa[x] == x)return x;
	return fa[x] = find(fa[x]);
}

find(a) != find(b)就合并两点，相等说明两点位于一个集合

没了。。

带权并查集

普通的并查集只能判断两点是否在同一集合，适用范围比较小

带权并查集同时维护每个结点到其祖先的距离 ，或者说将祖先看成根，维护每个点的深度dep，代码实现就是在路径压缩的时候，维护一个dep[]数组，递归到根结点后回来的时候不断dep[x] += dep[fa[x]]，同时让fa[x] = 祖先。

代码

inline int find(int x){
    if(fa[x] == x) return x;
    int last = fa[x];
    fa[x] = find(fa[x]);//一直递归到根
    dep[x] += dep[last];
    return fa[x];
}

在进行合并操作的时候，让右结点当根，dep[左] = dep[右]+1即可

当发现find(x) == find(y)时，说明出现了环，环的边数为dep[x] + dep[y] + 1，如图

 还有一个小细节是dep的更新问题，我们只让dep[x] = dep[y] + 1，是更新的所查询点的dep

---

P2661 [NOIP2015 提高组] 信息传递​​​​​​​

#include<bits/stdc++.h>
using namespace std;
int n, ans = 1e9, dep[210000],fa[210000];
inline int find(int x){
    if(fa[x] == x) return x;
    int last = fa[x];
    fa[x] = find(fa[x]);//一直递归到根
    dep[x] += dep[last];
    return fa[x];
}
int main(){
    cin >> n;
    for(int i = 1;i <= n;i++) fa[i] = i;
    for(int i = 1;i <= n;i++){
        int u = i, v;
        cin >> v;
        int a = find(u), b = find(v);
        if(a != b){
            fa[a] = b;
            dep[u] = dep[v] + 1;
        }
        else ans = min(ans, dep[u] + dep[v] + 1);
    }
    cout << ans;
    return 0;
}