70分TLE求助,玄关
查看原帖
70分TLE求助,玄关
574745
littlesnake楼主2024/1/27 19:24

rt.

代码:

#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 update(int root){
	t[root].size=t[t[root].left].size+t[t[root].right].size+t[root].num;
}
int get(int x,int root){
	if(root){
		if(x<t[root].value)return get(x,t[root].left);
		if(x>t[root].value)
			return get(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(){
	scanf("%d",&n);
	int num=0;
	t[root=++cnt]=Node(0,0,1,2147483647);
	while(n--){
		cin>>opt>>x;
		num++;
		if(opt==1)printf("%d\n",get(x,root));
		else if(opt==2)printf("%d\n",kth(x,root));
		else if(opt==3)printf("%d\n",kth(get(x,root)-1,root));
		else if(opt==4)printf("%d\n",kth(get(x+1,root),root));
		else num--,insert(x,root);
	}
	return 0;
}

2024/1/27 19:24
加载中...