bzoj2733(永无乡 支持其中的两个联通快合并 问某个联通快第k小 线段树合并模板)

题目
代码在bzoj T了 洛谷A了。。妈的不知道为什么。线段树合并裸体。。

#include<cstdio>
using namespace std;
const int N=3e5+5;
struct tree{int l,r,sum;}t[N*70];
int root[N],sz=0;
void build(int &x,int l,int r,int val){
    x=++sz;t[x].sum=1;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(val<=mid) build(t[x].l,l,mid,val);
    else build(t[x].r,mid+1,r,val);
    t[x].sum=t[t[x].l].sum+t[t[x].r].sum;
}
int query(int x,int l,int r,int k){
    if(l==r) return l;
    int mid=(l+r)>>1;
    if(t[t[x].l].sum>=k) return query(t[x].l,l,mid,k);
    return query(t[x].r,mid+1,r,k-t[t[x].l].sum);
}
int rk[N],sa[N],f[N];
inline int seek(int x){return (x==f[x])?(x):(f[x]=seek(f[x]));}
void combine(int &x,int y){
    if(!x||!y) {x=x+y;return;}
    t[x].sum+=t[y].sum;
    combine(t[x].l,t[y].l);
    combine(t[x].r,t[y].r);
}
char s[3];
int main(){
    int n,m;scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i){
        f[i]=i;
        scanf("%d",&rk[i]),sa[rk[i]]=i;
        build(root[i],1,n,rk[i]);
    }
    for(int i=1;i<=m;++i){
        int x,y;scanf("%d%d",&x,&y);
        x=seek(x),y=seek(y);
        if(x==y) continue;
        f[y]=x,combine(root[x],root[y]);
    }
    int q;scanf("%d",&q);
    while(q--){
        scanf("%s",s);
        if(s[0]=='B'){
            int x,y;scanf("%d%d",&x,&y);
            x=seek(x),y=seek(y);
            if(x==y) continue;
            f[y]=x,combine(root[x],root[y]);
        }else{
            int x,k;scanf("%d%d",&x,&k);x=seek(x);
            printf("%d\n",(t[root[x]].sum>=k)?(sa[query(root[x],1,n,k)]):(-1));
        }
    }
}