链式并查集合并(裸板)

对于操作:将一段元素合并为同类。

在合并 \([l,r]\) 这一段数的时候,其实有很多数本来就在一个并查集里。我们只需要合并若干个还没有合并的并查集,而不需要从 \(l\)\(r\) 一个一个合并。因为只要合并了这几个并查集,效果等价于把 \([l,r]\) 直接合并了。

实现方法:每次记录一个 \(nxt[i]\) 表示第 \(i\)个点后面第一个没有和 \(i\) 合并的点。每次合并的时候直接 $i=nxt[i] $处理即可,最后把 $[l,r] $中所有的 nxt[i] 都赋值为 \(nxt[r]\)。这样的话,两个不同的并查集在这一操作中至多合并一次。
https://codeforces.com/problemset/problem/566/D

#include<bits/stdc++.h>
#define endl '\n'
#define lowbit(x) (x&-x)
using namespace std;
const double pi=acos(-1);
const int N=2e5+5;
int fa[N],nx[N];

int find(int i){
	if(i!=fa[i]) fa[i]=find(fa[i]);//路径压缩
    return fa[i];
}

void join(int u,int v){
    u=find(u);
    v=find(v);
    if(u==v) return;
    if(u<v) swap(u,v);
    fa[v]=u;
}

void solve(){
    int n,q;cin>>n>>q;
    for(int i=1;i<=n+1;i++){
        fa[i]=i;
        nx[i]=i+1;
    }
    while(q--){
        int op;cin>>op;
        int x,y;cin>>x>>y;
        if(op==3){
            x=find(x);
            y=find(y);
            if(x==y) cout<<"YES"<<endl;
            else cout<<"NO"<<endl;
        }
        if(op==1){
            join(x,y);
        }
        if(op==2){
            int ty=find(y);
            for(int i=x;i<=y;){
                int t=i;
                i=nx[i];
                fa[find(t)]=ty;
                nx[t]=nx[y];
            }
        }
    }
}

signed main(){
    ios::sync_with_stdio(false);cin.tie(nullptr);
    int t=1;
    //cin>>t;
    while(t--) solve();
    return 0;
}



posted on 2024-11-06 18:51  TaopiTTT  阅读(16)  评论(0编辑  收藏  举报