问题描述

BZOJ2733

LG3224

题解

对于每个结点建立一棵权值线段树。

查询操作就去查询第 \(k\) 大，合并操作就合并两颗权值线段树。

并查集维护连通性。

同时 STO hkk,zcr,zzk

\(\mathrm{Code}\)

#include<bits/stdc++.h>
using namespace std;

template <typename Tp>
void read(Tp &x){
    x=0;char ch=1;int fh;
    while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();
    if(ch=='-') ch=getchar(),fh=-1;
    else fh=1;
    while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    x*=fh;
}

template <typename Tp>
void fr(Tp &x){
    char ch=1;
    while(ch!='Q'&&ch!='B') ch=getchar();
    if(ch=='Q') x=1;
    else x=2;
}

const int maxn=100003;

int n,m,T;
int rk[maxn];
int pre[maxn];

int rt[maxn],L[maxn*22],R[maxn*22],cnt;
int val[maxn*22],fa[maxn];

void pushup(int x){
    val[x]=val[L[x]]+val[R[x]];
}

int insert(int x,int l,int r,int p,int v){
    if(!x) x=++cnt;
    if(l==r){
        val[x]+=v;
        return x;
    }
    int mid=(l+r)>>1;
    if(p<=mid) L[x]=insert(L[x],l,mid,p,v);
    else R[x]=insert(R[x],mid+1,r,p,v);
    pushup(x);
    return x;
}

int merge(int p,int q,int l,int r){
    if(!p||!q) return p+q;
    if(l==r){
        val[q]+=val[p];return q;
    }
    int mid=(l+r)>>1;
    L[q]=merge(L[p],L[q],l,mid);
    R[q]=merge(R[p],R[q],mid+1,r);
    pushup(q);
    return q;
}

int find(int x){
    return fa[x]==x?x:fa[x]=find(fa[x]);
}

int query(int x,int l,int r,int k){
    if(!x&&k) return -1;
    if(r-l+1<k) return -1;
    if(l==r) return l;
    int mid=(l+r)>>1;
    if(!L[x]&&!R[x]&&k) return -1;
    if(L[x]&&val[L[x]]>=k) return query(L[x],l,mid,k);
    else return query(R[x],mid+1,r,k-val[L[x]]);
}

void cz1(){
    int x,k;read(x);read(k);
    int xx=find(x);
    int ppp=query(rt[xx],1,100000,k);
    printf("%d\n",pre[ppp]);
}

void cz2(){
    int x,y;read(x);read(y);
    int xx=find(x),yy=find(y);
    if(xx==yy) return;
    fa[xx]=yy; 
    merge(rt[xx],rt[yy],1,100000);
}

int main(){
    read(n);read(m);
    for(int i=1;i<=n;i++) read(rk[i]),pre[rk[i]]=i;
    for(int i=1;i<=n;i++) rt[i]=++cnt;
    for(int i=1;i<=n;i++){
        fa[i]=i;
        insert(rt[i],1,100000,rk[i],1);
    }
    
//  #define debug1
    
    #ifdef debug1
        for(int i=1;i<=n;i++)
            printf("%d %d\n",query(rt[i],1,100000,1),query(rt[i],1,100000,2));
    #endif
    
    for(int i=1,x,y;i<=m;i++){
        read(x);read(y);
        int xx=find(x),yy=find(y);
        if(xx==yy) continue;
        fa[xx]=yy;
        merge(rt[xx],rt[yy],1,100000);
    }
    int T,op;read(T);
    while(T--){
        fr(op);
        if(op==1) cz1();
        else cz2();
    }
    return 0;
}