并查集+路径压缩模板

并查集模板题链接
没啥好说的上图:
在这里插入图片描述
图示意思形象点说就是有两个联盟c联盟老大是c,f联盟老大是f,假设f打败了c,那c就带领它的小弟们加入了f。
具体代码:

        if(bc(b)!=bc(c)){
            head[bc(c)]=bc(b);//将c所在集合并入b所在集合
        }
int bc(int a){//递归寻找所查节点根节点
    if(head[a]==0) return a;
    return bc(head[a]);//如果当前节点不为根节点则继续递归
}

但不幸的是这样做很容易T,形象点说可以拿古代报信的兵来说本来眼前就是战机但因为要上报皇帝才能出兵,但回去报信一来一回花费太多时间贻误了战机导致灭国,所以我们应该想个办法使得报信兵能直接上报皇帝,让小弟们能直接联系大哥,这就是路径压缩
附代码:

int bc(int a){//递归寻找所查节点根节点
    if(head[a]==0) return a;
    return head[a]=bc(head[a]);//路径压缩(就是记忆化。。。)将每个节点直接与根节点相连
}

只需要在普通并查集上小小修改一下就行了。
最后附完整代码:

#include <bits/stdc++.h>
using namespace std;
#define int long long
int head[200005];
int bc(int a){//递归寻找所查节点根节点
    if(head[a]==0) return a;
    return head[a]=bc(head[a]);//路径压缩(就是记忆化。。。)将每个节点直接与根节点相连
}
void chaxun(int a){
    if(a==1){
        int b,c; cin>>b>>c;
        if(bc(b)!=bc(c)){
            head[bc(c)]=bc(b);//将c所在集合并入b所在集合
        }
    }
    if(a==2){
        int b,c; cin>>b>>c;
        if(bc(b)!=bc(c)){
            cout<<"N"<<endl;
        }
        else cout<<"Y"<<endl;
    }
}
signed main(){
    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    int n,m; cin>>n>>m;
    for(int i=0;i<m;i++){
        int a; cin>>a;
        chaxun(a);
    }
}

发布了28 篇原创文章 · 获赞 8 · 访问量 1579

猜你喜欢

转载自blog.csdn.net/chineseherofeng/article/details/105051658