RT,昨晚 AT 的 E 题偷懒复制自己这题的代码,然后喜提 WA。
输入:
1 1
1
1 1
1
Q 1 2
输出:
-1
我的输出:1
是线段树上二分的问题,如果 l==r 就直接 return 了,在只有 1 个元素时会出错。
被叉的代码:
#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,m;
int a[100005],rt[100005],pos[100005];
struct BCJ{
int f[100005];
void init(){
for(int i=0;i<=100000;i++){
f[i]=i;
}
}
int fd(int x){
if(f[x]==x) return x;
return f[x]=fd(f[x]);
}
void merge(int x,int y){
int fx=fd(x),fy=fd(y);
if(fx!=fy) f[fx]=fy;
}
}qset;
struct node{
int sum,lc,rc;
};
struct sgt{
int cnt=0;
node t[6400005];
void newnode(int &p){
p=++cnt;
t[p]={0,0,0};
}
void update(int &p,int l,int r,int pos,int val){
if(!p) newnode(p);
if(l==r){
t[p].sum+=val;
return;
}
int mid=(l+r)>>1;
if(mid>=pos){
update(t[p].lc,l,mid,pos,val);
}else{
update(t[p].rc,mid+1,r,pos,val);
}
t[p].sum=t[t[p].lc].sum+t[t[p].rc].sum;
}
int query(int p,int l,int r,int k){
if(!p) return -1;
if(l==r){//这里有问题
return l;
}
int mid=(l+r)>>1,res=0;
if(t[t[p].lc].sum>=k){
return query(t[p].lc,l,mid,k);
}else if(t[t[p].rc].sum>=k-t[t[p].lc].sum){
return query(t[p].rc,mid+1,r,k-t[t[p].lc].sum);
}else{
return -1;
}
}
int kth(int rt,int k){
// double t=clock();
// cout<<"kth start!\n";
int res=query(rt,1,n,k);
// cout<<"kth finish:"<<(double)clock()-t<<"\n";
return res;
}
int merge(int a,int b,int l,int r){
if(!a) return b;
if(!b) return a;
if(l==r){
t[a].sum+=t[b].sum;
return a;
}
int mid=(l+r)/2;
t[a].lc=merge(t[a].lc,t[b].lc,l,mid);
t[a].rc=merge(t[a].rc,t[b].rc,mid+1,r);
t[a].sum=t[t[a].lc].sum+t[t[a].rc].sum;
return a;
}
}segt;
signed main(){
// freopen("1.in","r",stdin);
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>a[i];
pos[a[i]]=i;
segt.newnode(rt[i]);
}
for(int i=1;i<=n;i++){
segt.update(rt[i],1,n,a[i],1);
}
qset.init();
for(int i=1;i<=m;i++){
int a,b;
cin>>a>>b;
int fa=qset.fd(a),fb=qset.fd(b);
qset.merge(fa,fb);
if(fb!=fa) rt[fb]=segt.merge(rt[fb],rt[fa],1,n);
if(fb!=fb) rt[fb]=segt.merge(rt[fb],rt[fb],1,n);
}
int q;
cin>>q;
while(q--){
char c;
int a,b;
cin>>c>>a>>b;
if(c=='Q'){
int fa=qset.fd(a);
int ans=segt.kth(rt[fa],b);
if(ans!=-1) ans=pos[ans];
cout<<ans<<"\n";
}else{
int fa=qset.fd(a),fb=qset.fd(b);
qset.merge(fa,fb);
if(fb!=fa) rt[fb]=segt.merge(rt[fb],rt[fa],1,n);
if(fb!=fb) rt[fb]=segt.merge(rt[fb],rt[fb],1,n);
}
}
return 0;
}
/*
5 3
4 3 2 5 1
1 2
2 3
1 5
1
Q 2 4
*/