求助,样例没过
查看原帖
求助,样例没过
766106
Tr_Sup楼主2023/2/3 22:18
#include<bits/stdc++.h>
#define MAXN 100010 
using namespace std;
int n,root,cnt,opt,x;
struct node{
	int left,right,size,value,num;
	node(int l,int r,int s,int v)
		: left(l),right(r),size(s),value(v),num(1) {}
	node(){}
}t[MAXN]; 
inline void ud(int root){
	t[root].size=t[t[root].left].size+t[t[root].right].size+t[root].num;
}
int rank(int x,int root){
	if(root){
		if(x<t[root].value){
		return rank(x,t[root].left);
		}
		if(x>t[root].value){
			return rank(x,t[root].right)+t[t[root].left].size+t[root].num;
		}
		return t[t[root].left].size+t[root].num;
	} 
	return 1;
}
int kth(int x,int root){
	if(x<=t[t[root].left].size){
		return kth(x,t[root].left);
	}
	if(x<=t[t[root].left].size+t[root].num){
		return t[root].value;
	}
	return kth(x-t[t[root].left].size-t[root].num,t[root].right);
}
void insert(int x,int & root){
	if(x<t[root].value){
		if(!t[root].left){
			t[t[root].left=++cnt]=node(0,0,1,x);
		}
		else{
			insert(x,t[root].left);
		}
	}
	else if(x>t[root].value){
		if(!t[root].right){
			t[t[root].right=++cnt]=node(0,0,1,x);
		}
		else{
			insert(x,t[root].right);
		}
	}
	else{
		t[root].num++;
	}
	ud(root);
}
int main(){
	cin>>n;
	t[root=++cnt]=node(0,0,1,2147483674);
	while(n--){
		cin>>opt>>x;
		if(opt==1) cout<<rank(x,root)<<endl;
		else if(opt==2) cout<<kth(x,root)<<endl;
		else if(opt==3) cout<<kth(rank(x,root)-1,root)<<endl;
		else if(opt==4) cout<<kth(rank(x,root)+1,root)<<endl;
		else insert(x,root);
	}
	return 0;
}

书上的代码,样例没过,是哪里出了问题吗??

2023/2/3 22:18
加载中...