求解
查看原帖
求解
1055123
dmh0612楼主2024/7/14 16:47

cpp #include<bits/stdc++.h> 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[100010]; inline void update(int root){ t[root].size = t[t[root].left].size+t[t[root].right].size+t[root].num; } int rk(int x,int root){ if(root){ if(x<t[root].value) return rk(x,t[root].left); if(x>t[root].value) return rk(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(){ ios::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL); cin>>n; int num = 0; t[root = ++cnt] = Node(0,0,1,2147483647); while(n--){ cin>>opt>>x; num++; if(opt == 1) cout<<rk(x,root)<<'\n'; else if(opt == 2) cout<<kth(x,root)<<'\n'; else if(opt == 3) cout<<kth(rk(x,root)-1,root)<<'\n'; else if(opt == 4) cout<<kth(rk(x+1,root),root)<<'\n'; else num--,insert(x,root); } return 0; }

2024/7/14 16:47
加载中...