[BZOJ2733][HNOI2012]永无乡 线段树合并

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=2733

我们对每一个连通块建一棵以排名为键值的权值线段树，询问就可以用二分的方法。

然后对于块与块间加边，我们用并查集维护连通性，每次合并两个块的时候，因为线段树结构相同，可以直接合并就行了。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int inline readint(){
    int Num;char ch;
    while((ch=getchar())<'0'||ch>'9');Num=ch-'0';
    while((ch=getchar())>='0'&&ch<='9') Num=Num*10+ch-'0';
    return Num;
}
void outint(int x){
    if(x>=10) outint(x/10);
    putchar(x%10+'0');
}
int N,M,Q;
int a[100010],re[100010];
int Rt[100010],lc[2000010],rc[2000010],cnt=0;
int sum[2000010];
int fa[100010];
int inline Getfa(int x){
    return fa[x]==x?x:fa[x]=Getfa(fa[x]);
}
void Pushup(int &rt){
    sum[rt]=sum[lc[rt]]+sum[rc[rt]];
}
void Insert(int &rt,int l,int r,int x){
    if(!rt) rt=++cnt;
    if(l==r){
        sum[rt]=1;
        return;
    }
    int mid=l+r>>1;
    if(x<=mid) Insert(lc[rt],l,mid,x);
    else Insert(rc[rt],mid+1,r,x);
    Pushup(rt);
}
int Merge(int x,int y){
    if(!x||!y) return x+y;
    lc[x]=Merge(lc[x],lc[y]);
    rc[x]=Merge(rc[x],rc[y]);
    Pushup(x);
    return x;
}
int Qry(int rt,int l,int r,int x){
    if(l==r) return l;
    int mid=l+r>>1;
    if(sum[lc[rt]]>=x) return Qry(lc[rt],l,mid,x);
    else return Qry(rc[rt],mid+1,r,x-sum[lc[rt]]);
}
int main(){
    N=readint();
    M=readint();
    for(int i=1;i<=N;i++){
        a[i]=readint();
        Insert(Rt[i],1,N,a[i]);
        re[a[i]]=i;
    }
    for(int i=1;i<=N;i++) fa[i]=i;
    for(int i=1;i<=M;i++){
        int A=readint(),
            B=readint(),
            fA=Getfa(A),
            fB=Getfa(B);
        fa[fB]=fA;
        Rt[fA]=Merge(Rt[fA],Rt[fB]);
    }
    Q=readint();
    for(int i=1;i<=Q;i++){
        char opt[5];
        scanf("%s",opt);
        int x=readint(),
            y=readint();
        switch(opt[0]){
            case 'Q':
                x=Getfa(x);
                if(sum[Rt[x]]<y){
                    puts("-1");
                    break;
                }
                outint(re[Qry(Rt[x],1,N,y)]);
                putchar('
');
                break;
            case 'B':
                x=Getfa(x);
                y=Getfa(y);
                if(x==y) break;
                fa[y]=x;
                Rt[x]=Merge(Rt[x],Rt[y]);
                break;
        }
    }
    return 0;
}