Hihocoder 1337 (splay)

Problem 平衡树 SBT

题目大意

　　维护一个序列，支持两种操作。

　　操作一：插入一个数。

　　操作二：询问第k小的数。

解题分析

　　~~刷刷水题，再熟悉一下splay的基本操作。

　　ps：哇咔咔，有连续四天的假期了，好开心~~

参考程序

 #include <bits/stdc++.h>
 using namespace std;

 struct node{
     int v,sz;
     node *l,*r,*f;
     node(int v_=,int sz_=,node* f_=NULL,node* l_=NULL,node* r_=NULL)
     {
         v=v_; sz=sz_; l=l_; r=r_; f=f_;
     }
 }*rt;

 inline void pushup(node *x)
 {
     x->sz=;
     if (x->l) x->sz += x->l->sz;
     if (x->r) x->sz += x->r->sz;
 }
 void left(node* x,node* &rt)
 {
     node *y=x->f , *z=y->f;
     if (y==rt) rt=x; else
         if (y==z->l) z->l=x; else z->r=x;
     x->f=y->f; y->f=x; if (x->l) x->l->f=y; y->r=x->l; x->l=y;
     pushup(y); pushup(x);
 }
 void right(node* x,node* &rt)
 {
     node *y=x->f , *z=y->f;
     if (y==rt) rt=x; else
         if (y==z->l) z->l=x; else z->r=x;
     x->f=y->f; y->f=x; if (x->r) x->r->f=y; y->l=x->r; x->r=y;
     pushup(y); pushup(x);
 }
 void splay(node* x,node* &rt)
 {
     while (x!=rt)
     {
         node *y=x->f , *z=y->f;
         if (y==rt)
             if (x == y->l) right(x,rt); else left(x,rt);
         else
             if (y == z->l)
                 if (x == y->l) {right(y,rt);right(x,rt);}
                     else {left(x,rt);right(x,rt);}
             else
                 if (x == y->r) {left(y,rt);left(x,rt);}
                     else {right(x,rt);left(x,rt);}
     }
 }

 void insert(int v,node* &x,node* f)
 {
     if (x==NULL)
     {
         x=new node(v,,f);
         splay(x,rt);
         return;
     }
     if (v < x->v) insert(v,x->l,x); else insert(v,x->r,x);
 }

 int query(int k,node *x)
 {
     int num=x->l?x->l->sz:;
     if (k==num+) return x->v;
     if (k<num+) return query(k,x->l);
     return query(k-num-,x->r);
 }
 void search(node *x)
 {
     if (x==NULL) return;
     cout<<x->v<<" "<<x->sz;
     if (x->l) cout<<" lson:"<<x->l->v;
     if (x->r) cout<<" rson:"<<x->r->v;
     cout<<endl;
     if (x->l) search(x->l);
     if (x->r) search(x->r);

 }
 int main()
 {
     int n;
     rt=NULL;
     scanf("%d",&n);
     for (int i=;i<=n;i++)
     {
         //search(rt);
         char s[]; int x;
         scanf("%s%d",s,&x);
         if (s[]=='I') insert(x,rt,NULL); else cout<<query(x,rt)<<endl;
     }
 }