代码输入一行停 求助dalao
查看原帖
代码输入一行停 求助dalao
737406
shensteve楼主2022/8/11 22:41
#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(1),right(r),size(s),value(v),num(1) {}
	Node() {}
}t[MAXN];
inline void update(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++;
	update(root);	    
}
int main(){
    cin>>n;
    int num=0;
    t[root=++cnt]=Node(0,0,1,2147483647);
    while(n--){
    	cin>>opt>>x;
    	num++;
    	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+1,root),root)<<endl;
    	else num--,insert(x,root);
	}
	return 0;
}
2022/8/11 22:41
加载中...