链式并查集合并(裸板)
对于操作:将一段元素合并为同类。
在合并 \([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;
}