问题描述

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;
}
04-29 04:29