平衡树之treap luoguP3369

今天又复习了一遍treap，这题有前驱后继排名排位添加和删除等操作。

非常好写，虽然代码颇长但逻辑性很强。

 #include<bits/stdc++.h>
 using namespace std;
 const int N=;
 struct node
 {
     int l,r,w,v,size,heap;
 }t[N];
 int n,rt,cnt;
 void merge(int p)
 {
     t[p].size=t[t[p].l].size+t[t[p].r].size+t[p].w;
 }
 void rturn(int &p)
 {
     int ls=t[p].l;
     t[p].l=t[ls].r;
     t[ls].r=p;
     t[ls].size=t[p].size;
     merge(p);
     p=ls;
 }
 void lturn(int &p)
 {
     int rs=t[p].r;
     t[p].r=t[rs].l;
     t[rs].l=p;
     t[rs].size=t[p].size;
     merge(p);
     p=rs;
 }
 void insert(int &k,int x)
 {
     if(!k)
     {
         cnt++;
         k=cnt;
         t[k].w=t[k].size=;
         t[k].v=x;
         t[k].heap=rand();
         return;
     }
     t[k].size++;
     if(t[k].v==x)
     t[k].w++;
     else
     {
         if(t[k].v<x)
         {
             insert(t[k].r,x);
             if(t[t[k].r].heap>t[k].heap)
             lturn(k);
         }
         if(t[k].v>x)
         {
             insert(t[k].l,x);
             if(t[t[k].l].heap>t[k].heap)
             rturn(k);
         }
     }
 }
 int tmp;
 void querypre(int k,int x)
 {
     if(!k)return;
     if(x>t[k].v)
     {
         tmp=t[k].v;querypre(t[k].r,x);
     }
     else querypre(t[k].l,x);
 }
 void querylas(int k,int x)
 {
     if(!k)return;
     if(x<t[k].v)
     {
         tmp=t[k].v;querylas(t[k].l,x);
     }
     else querylas(t[k].r,x);
 }
 void del(int &k,int x)
 {
     if(!k)return;
     if(t[k].v==x)
     {
         if(t[k].w>)
         {
             t[k].w--;t[k].size--;
         }
         else if(!t[k].l||!t[k].r)k=t[k].l+t[k].r;
         else
         {
             if(t[t[k].l].heap>t[t[k].r].heap)
             {
                 rturn(k);
                 del(k,x);
             }
             else
             {
                 lturn(k);
                 del(k,x);
             }
         }
     }
     else
     {
         t[k].size--;
         if(x>t[k].v)
         del(t[k].r,x);
         else
         del(t[k].l,x);
     }
 }
 int queryrank(int k,int x)
 {
     if(k==)return ;
     if(t[k].v==x)
     return t[t[k].l].size+;
     else
     {
         if(x>t[k].v)return t[t[k].l].size+t[k].w+queryrank(t[k].r,x);
         else return queryrank(t[k].l,x);
     }
 }
 int querysum(int k,int x)
 {
     if(k==)return ;
     if(x<=t[t[k].l].size)return querysum(t[k].l,x);
     else
     {
         if(x>t[t[k].l].size+t[k].w)return querysum(t[k].r,x-t[t[k].l].size-t[k].w);
         else return t[k].v;
     }
 }
 int main()
 {
     scanf("%d",&n);
     for(int i=;i<=n;++i)
     {
         int f,x;
         scanf("%d%d",&f,&x);
         if(f==)insert(rt,x);
         else if(f==)del(rt,x);
         else if(f==)printf("%d\n",queryrank(rt,x));
         else if(f==)printf("%d\n",querysum(rt,x));
         else if(f==){
             querypre(rt,x);printf("%d\n",tmp);
         }
         else{
             querylas(rt,x);printf("%d\n",tmp);
         }
     }
     return ;
 }