P5076 MLE 求救
  • 板块灌水区
  • 楼主珂学至上
  • 当前回复11
  • 已保存回复11
  • 发布时间2020/8/4 23:08
  • 上次更新2023/11/6 21:16:59
查看原帖
P5076 MLE 求救
334884
珂学至上楼主2020/8/4 23:08

蒟蒻五个点都MLE了

哪里有不对的地方,还望大佬告知QAQ

#include<bits/stdc++.h>
#define MAXN 100050
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 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)
        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;
}

TOT

2020/8/4 23:08
加载中...