曾经的曾经,一名年轻的OIer以为这是一道可合并平衡树,然后暗骂了一句出题人。后来,他知道线段树居然是可以合并的。再后来,他又知道了线段树居然是可以动态开点的。年轻的少年回头一看发现:哇,原来这题是傻逼题。
没错,这个东西可以轻松线段树并查集搞掉,合并的时候由于是动态开的点,所以时间保障在log以内,之后权值线段树查询的时候分治搞一下就行了。
傻逼少年的代码
#include <cstdio>
using namespace std;
const int Maxn=2e6+5;
int n,m,q,a,b,cnt=0;
int w[Maxn],f[Maxn];
int root[Maxn],lson[Maxn],rson[Maxn],sum[Maxn],id[Maxn];
int fd(int x){
if(x==f[x]) return f[x];
else return f[x]=fd(f[x]);
}
void build(int &k,int l,int r,int val){
if(!k) k=++cnt;
if(l==r){ sum[k]=1; return; }
int mid=(l+r)>>1;
if(val<=mid) build(lson[k],l,mid,val);
else build(rson[k],mid+1,r,val);
sum[k]=sum[lson[k]]+sum[rson[k]];
}
int merge(int x,int y){
if(!x) return y;
if(!y) return x;
lson[x]=merge(lson[x],lson[y]);
rson[x]=merge(rson[x],rson[y]);
sum[x]=sum[lson[x]]+sum[rson[x]];
return x;
}
int query(int o,int l,int r,int x){
if(l==r) return l;
int mid=(l+r)>>1;
if(sum[lson[o]]>=x) return query(lson[o],l,mid,x);
else return query(rson[o],mid+1,r,x-sum[lson[o]]);
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i) scanf("%d",&w[i]);
for(int i=1;i<=n;++i) f[i]=i;
for(int i=1;i<=m;++i){
scanf("%d%d",&a,&b);
int fx=fd(a),fy=fd(b);
if(fx!=fy) f[fx]=fy;
}
for(int i=1;i<=n;++i){
build(root[fd(i)],1,n,w[i]);
id[w[i]]=i;
}
scanf("%d",&q);
char s[3];
for(int i=1;i<=q;++i){
scanf("%s%d%d",s,&a,&b);
if(s[0]=='Q'){
int p=root[fd(a)];
if(sum[p]<b) puts("-1");
else{
int ans=query(p,1,n,b);
printf("%d\n",id[ans]);
}
}
else{
int fx=fd(a),fy=fd(b);
if(fx!=fy){
f[fx]=fy;
root[fy]=merge(root[fx],root[fy]);
}
}
}
return 0;
}