bzoj 3224 普通平衡树

Treap做法版。。。。

/**************************************************************
    Problem: 3224
    User: Clare
    Language: C++
    Result: Accepted
    Time:276 ms
    Memory:3620 kb
****************************************************************/
 
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <queue>
#include <vector>
using namespace std;
  
#define N 100010
#define INF 0x7fffffff
  
int n,m,root,tot,Ans;
struct Node{
    int L,R,size,rnd,value,w;
}t[N];
  
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
  
void Pushup(int k)
{
    t[k].size=t[t[k].L].size+t[t[k].R].size+t[k].w;
}
  
void Right_rotate(int &k)
{
    int tmp=t[k].L;t[k].L=t[tmp].R;t[tmp].R=k;
    t[tmp].size=t[k].size;Pushup(k);k=tmp;
}
  
void Left_rotate(int &k)
{
    int tmp=t[k].R;t[k].R=t[tmp].L;t[tmp].L=k;
    t[tmp].size=t[k].size;Pushup(k);k=tmp;
}
  
void Insert(int &k,int x)
{
    if(k==0)
    {
        k=++tot;t[k].value=x;t[k].size=t[k].w=1;
        t[k].rnd=rand();return;
    }
    t[k].size++;
    if(t[k].value==x)
        t[k].w++;
    else if(x<t[k].value)
    {
        Insert(t[k].L,x);
        if(t[t[k].L].rnd<t[k].rnd)Right_rotate(k);
    }
    else
    {
        Insert(t[k].R,x);
        if(t[t[k].R].rnd<t[k].rnd)Left_rotate(k);
    }
}
  
void Del(int &k,int x)
{
    if(k==0)return;
    if(t[k].value==x)
    {
        if(t[k].w>1)
        {
            t[k].w--;t[k].size--;return;
        }
        if(t[k].L*t[k].R==0)k=t[k].L+t[k].R;
        else if(t[t[k].L].rnd<t[t[k].R].rnd)
            Right_rotate(k),Del(k,x);
        else Left_rotate(k),Del(k,x);
    }
    else if(x>t[k].value)
        t[k].size--,Del(t[k].R,x);
    else t[k].size--,Del(t[k].L,x);
}
  
int Query(int k,int x)
{
    if(k==0)return 0;
    int l=t[k].L,r=t[k].R;
    if(x<=t[l].size)
        return Query(l,x);
    else if(x<=t[l].size+t[k].w)
        return t[k].value;
    else return Query(r,x-t[l].size-t[k].w);
}
  
int Find_rank(int k,int x)
{
    if(k==0)return 0;
    if(t[k].value==x)
        return t[t[k].L].size+1;
    else if(x<t[k].value)
        return Find_rank(t[k].L,x);
    else return Find_rank(t[k].R,x)+t[k].w+t[t[k].L].size;
}
  
void Query_be(int k,int x)
{
    if(k==0)return;
    if(t[k].value<x)
    {
        Ans=k;Query_be(t[k].R,x);
    }
    else Query_be(t[k].L,x);
}
  
void Query_af(int k,int x)
{
    if(k==0)return;
    if(t[k].value>x)
    {
        Ans=k;Query_af(t[k].L,x);
    }
    else Query_af(t[k].R,x);
}
  
int main()
{
    n=read();
    for(int i=1;i<=n;i++)
    {
        int kind=read(),x=read();
        if(kind==1)Insert(root,x);
        else if(kind==2)Del(root,x);
        else if(kind==3)printf("%d\n",Find_rank(root,x));
        else if(kind==4)printf("%d\n",Query(root,x));
        else if(kind==5)
        {
            Ans=0;Query_be(root,x);
            printf("%d\n",t[Ans].value);
        }
        else if(kind==6)
        {
            Ans=0;Query_af(root,x);
            printf("%d\n",t[Ans].value);
        }
    }
    return 0;
}