线段树合并+并查集 || BZOJ 2733: [HNOI2012]永无乡 || Luogu P3224 [HNOI2012]永无乡

题面：P3224 [HNOI2012]永无乡

题解：

随便写写

代码：

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 using namespace std;
 const int maxn=(1e5)+,maxm=maxn,maxq=(3e5)+;
 int N,M,W[maxn],fa[maxn],root[maxn],U,V,num_treenode=,Q,ans,belong[maxn];
 char c;
 inline int getf(int x){
     if(fa[x]==x)return fa[x];
     fa[x]=getf(fa[x]);
     return fa[x];
 }
 struct Tree{int lc,rc,l,r,cnt;}tr[maxn*];
 inline void Pushup(int x){
     int lc=tr[x].lc,rc=tr[x].rc;
     tr[x].cnt=tr[lc].cnt+tr[rc].cnt;
     return;
 }
 inline void Build(int x,int l,int r,int q){
     tr[x].l=l;tr[x].r=r;int mid=(l+r)>>;
     if(l==r){
         tr[x].cnt=;
         return;
     }
     if(q<=mid)Build(tr[x].lc=++num_treenode,l,mid,q);
     else Build(tr[x].rc=++num_treenode,mid+,r,q);
     Pushup(x);
     return;
 }
 inline int Merge(int u,int v){
     if(!u)return v;
     if(!v)return u;
     int l=tr[u].l,r=tr[u].r;
     if(l==r){
         tr[u].cnt+=tr[v].cnt;
         return u;
     }
     tr[u].lc=Merge(tr[u].lc,tr[v].lc);
     tr[u].rc=Merge(tr[u].rc,tr[v].rc);
     Pushup(u);
     return u;
 }
 inline int Query(int x,int k){
     int l=tr[x].l,r=tr[x].r,lc=tr[x].lc,rc=tr[x].rc;
     if(l==r)return l;
     if(tr[lc].cnt>=k)return Query(lc,k);
     else return Query(rc,k-tr[lc].cnt);
 }
 int main(){
     scanf("%d%d",&N,&M);
     for(int i=;i<=N;i++){
         scanf("%d",&W[i]);
         fa[i]=i;
         belong[W[i]]=i;
     }
     for(int i=;i<=N;i++)
         Build(root[i]=++num_treenode,,N,W[i]);
     for(int i=;i<=M;i++){
         scanf("%d%d",&U,&V);
         int fa1=getf(U),fa2=getf(V);
         if(fa1!=fa2){
             Merge(root[fa1],root[fa2]);
             fa[fa2]=fa1;
         }
     }
     scanf("%d",&Q);
     while(Q--){
         c=getchar();
         while(c!='Q'&&c!='B')c=getchar();
         scanf("%d%d",&U,&V);
         if(c=='Q'){
             int f=getf(U);
             ans=Query(root[f],V);
             if(ans==)ans=-;else ans=belong[ans];
             printf("%d\n",ans);
         }
         else{
             int fa1=getf(U),fa2=getf(V);
             if(fa1!=fa2){
                 Merge(root[fa1],root[fa2]);
                 fa[fa2]=fa1;
             }
         }
     }
     return ;
 }

---

By:AlenaNuna