把自己叉了
查看原帖
把自己叉了
377969
george0929楼主2024/9/22 07:38

RT,昨晚 AT 的 E 题偷懒复制自己这题的代码,然后喜提 WA。

输入:

1 1
1
1 1
1
Q 1 2

输出:

-1

我的输出:1

是线段树上二分的问题,如果 l==r 就直接 return 了,在只有 11 个元素时会出错。

被叉的代码:

#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
*/
2024/9/22 07:38
加载中...