bzoj2733 / P3224 [HNOI2012]永无乡（并查集+线段树合并）

[HNOI2012]永无乡

每个联通块的点集用动态开点线段树维护

并查集维护图

合并时把线段树也合并就好了。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<cctype>
 #define re register
 #define gc getchar
 using namespace std;
 void chread(char &x){
     char c=gc();
     while(!isupper(c)) c=gc();
     x=c;
 }
 void read(int &x){
     char c=gc();x=; bool f=;
     while(!isdigit(c)) f=(f&&c!='-'),c=gc();
     while(isdigit(c)) x=(x<<)+(x<<)+(c^),c=gc();
     x=f?x:-x;
 }
 void swap(int &a,int &b){a^=b;b^=a;a^=b;}
 #define N 100002
 #define M 600002
 struct edge{int u,v;bool del;}b[M];
 struct node{
     int sum,lc,rc;
 }c[M<<];
 int n,m,t,u,val[N],fa[N],rt[N],ttp,id[N];
 double ans;
 
 void update(int &o,int l,int r,int v,int k){//插入
     if(!o) o=++u;
     if(l==r) {c[o].sum+=k; return;}
     int mid=l+((r-l)>>);
     if(v<=mid) update(c[o].lc,l,mid,v,k);
     else update(c[o].rc,mid+,r,v,k);
     c[o].sum=c[c[o].lc].sum+c[c[o].rc].sum;
 }
 void merge(int &o,int pr){//合并
     if(!o||!pr) {o+=pr;return;}
     c[o].sum+=c[pr].sum;
     merge(c[o].lc,c[pr].lc);
     merge(c[o].rc,c[pr].rc);
 }
 int query(int o,int l,int r,int k){//询问
     if(l==r) return l;
     int mid=l+((r-l)>>);
     if(k<=c[c[o].lc].sum) return query(c[o].lc,l,mid,k);
     else return query(c[o].rc,mid+,r,k-c[c[o].lc].sum);
 }
 int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
 void uni(int x,int y){//并查集合并
     int r1=find(x),r2=find(y);
     if(r1!=r2){
         merge(rt[r1],rt[r2]);
         fa[r2]=r1;
     }
 }
 int ask(int x,int k){
     int ffa=find(x);
     return k>c[rt[ffa]].sum ?-:id[query(rt[ffa],,n,k)];
 }
 int main(){
     char opt; int q1,q2;
     read(n); read(m);
     for(re int i=;i<=n;++i){
         read(val[i]),id[val[i]]=i,fa[i]=i;
         update(rt[i],,n,val[i],);
     }
     for(re int i=;i<=m;++i) read(q1),read(q2),uni(q1,q2);
     read(t);
     for(re int i=;i<=t;++i){
         chread(opt); read(q1); read(q2);
         switch(opt){
             case 'B':{uni(q1,q2); break;}
             case 'Q':{printf("%d\n",ask(q1,q2));  break;}
         }
     }
     return ;
 }