Dynamic Query System 基本SBT

SBT神器啊，川大校赛出了个基本的SBT，他们搞的数据有点恶心，写了好几次都没过掉

http://222.196.33.254/oj/showproblem?problem_id=1716 这个是我们学院上的地址，不过好像只有教育网可以登

感觉SBT就是size一个信息就可以解决n多动态查询问题了

这个可以用来当做模板，自己写的代码可读性还是比较高

/* * File: main.cpp * Author: Mi * * Created on 2011年4月13日, 下午7:51 */ #include <cstdlib> #include <stdio.h> #include <string.h> #include <algorithm> #include <cassert> #define N 500005 using namespace std; /* * */ int tol=0; struct SBT { int left,right; int key; int size; void init() { left=right=0; size=1; } }T[N]; void R_Rotate(int &t)//右旋 { int k=T[t].left; T[t].left=T[k].right; T[k].right=t; T[k].size=T[t].size; T[t].size=T[T[t].left].size+T[T[t].right].size+1; t=k; return ; } void L_Rotate(int &t)//左旋 { int k=T[t].right; T[t].right=T[k].left; T[k].left=t; T[k].size=T[t].size; T[t].size=T[T[t].left].size+T[T[t].right].size+1; t=k; } void Maintain(int &t,bool flag)//维护，SBT精华之所在 { if(flag==false) { if(T[T[T[t].left].left].size>T[T[t].right].size) R_Rotate(t); else if(T[T[T[t].left].right].size>T[T[t].right].size) { L_Rotate(T[t].left); R_Rotate(t); } else return ; } else { if(T[T[T[t].right].right].size>T[T[t].left].size) L_Rotate(t); else if(T[T[T[t].right].left].size>T[T[t].left].size) { R_Rotate(T[t].right); L_Rotate(t); } else return ; } Maintain(T[t].left,false); Maintain(T[t].right,true); Maintain(t,false); Maintain(t,true); } void Insert(int &t,int v)//插入 { if(t==0) { t=++tol; T[t].init(); T[t].key=v; } else { T[t].size++; if(v<T[t].key) Insert(T[t].left,v); else Insert(T[t].right,v); Maintain(t,v>=T[t].key); } } int Delete(int &t,int v)//删除 { if(!t) return 0; T[t].size--; if(v==T[t].key||v<T[t].key&&!T[t].left||v>T[t].key&&!T[t].right) { if(T[t].left&&T[t].right) { int p=Delete(T[t].left,v+1); T[t].key=T[p].key; return p; } else { int p=t; t=T[t].left+T[t].right; return p; } } else return Delete(v<T[t].key?T[t].left:T[t].right,v); } int Find_k(int t,int k)//找出第k大数 { if(k<=T[T[t].left].size) return Find_k(T[t].left,k); else if(k>T[T[t].left].size+1) return Find_k(T[t].right,k-T[T[t].left].size-1); return T[t].key; } int Getmin(int t)//取最小值 { while(T[t].left) t=T[t].left; return t; } int Getmax(int t)//取最大值 { while(T[t].right) t=T[t].right; return t; } int Rank(int t,int key)//排名其实就是它的左子树的size+1 { if(t==0) return 0; if(key<=T[t].key) return Rank(T[t].left,key); else return T[T[t].left].size+1+Rank(T[t].right,key); } int Exist(int t,int x)//判断这个节点是否存在 { if(t==0) return 0; if(x<T[t].key) return Exist(T[t].left,x); else if(x==T[t].key) return 1; else return Exist(T[t].right,x); } int Count(int t,int x)//统计出现次数 { if(!Exist(t,x)) return 0; else return Rank(t,x+1)-Rank(t,x); } int Pred(int t,int v)//返回比v小的最大的数 { if(t==0) return v; else if(v>T[t].key) { int ret=Pred(T[t].right,v); if(ret==v) return T[t].key; return ret; } else return Pred(T[t].left,v); } int Succ(int t,int v)//返回比v大的最小的数 { if(t==0) return v; else if(v<T[t].key) { int ret=Succ(T[t].left,v); if(ret==v) return T[t].key; return ret; } else return Succ(T[t].right,v); } int main(int argc, char** argv) { int t,q; // freopen("j1.in","r",stdin); // freopen("out.out","w",stdout); scanf("%d",&t); while(t--) { int root=0; tol=0; scanf("%d",&q); while(q--) { char str[5]; int x; scanf("%s",str); if(str[0]=='I') { scanf("%d",&x); Insert(root,x); } else if(str[0]=='R') { scanf("%d",&x); if(Exist(root,x)) Delete(root,x); } else if(str[0]=='S') printf("%d/n",T[root].size); else if(str[0]=='L') { scanf("%d",&x); printf("%d/n",Rank(root,x)); } else if(str[0]=='W') { scanf("%d",&x); if(x<1||x>T[root].size) printf("-1/n"); else printf("%d/n",Find_k(root,x)); } else if(str[0]=='C') { scanf("%d",&x); printf("%d/n",Count(root,x)); } else if(str[1]=='I') { if(root==0) puts("-1"); else printf("%d/n",T[Getmin(root)].key); } else if(str[1]=='A') { if(root==0) puts("-1"); else printf("%d/n",T[Getmax(root)].key); } /* else if(str[0]=='P') { scanf("%d",&x); printf("%d/n",Pred(root,x)); } else if(str[1]=='U') { scanf("%d",&x); printf("%d/n",Succ(root,x)); }*/ } } return 0; }