[bzoj 3224]手写treap

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=3224

bzoj不能用time(0)，看到这个博客才知道，我也RE了好几发……

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

struct Node
{
    int val,pri,sz,cnt;
    Node* ch[];
    int cmp(int x) const
    {
        if (x==val) return -;
        return x<val?:;
    }
    void push_up()
    {
        sz=cnt+ch[]->sz+ch[]->sz;
    }
}*tree,*null;

void rotate(Node* &node,int d)
{
    Node *k=node->ch[d];
    node->ch[d]=k->ch[d^];
    k->ch[d^]=node;
    node=k;
}

void insert(Node* &node,int x)
{
    if (node==null)
    {
        node=new Node();
        node->ch[]=null;
        node->ch[]=null;
        node->val=x;
        node->cnt=;
        node->pri=rand();
        node->sz=;
    }
    else
    {
        int d=node->cmp(x);
        if (d==-)
        {
            node->cnt++;
            node->sz++;
        }
        else
        {
            insert(node->ch[d],x);
            if (node->pri < node->ch[d]->pri)
            {
                rotate(node,d);
                node->ch[d^]->push_up();
            }
            node->push_up();
        }
    }
}

void remove(Node* &node,int x)
{
    if (node==null) return;
    int d=node->cmp(x);
    if (d==-)
    {
        if (node->cnt>)
        {
            node->cnt--;
            node->sz--;
        }
        else
        {
            if (node->ch[]==null)
            {
                Node *tmp=node;
                node=node->ch[];
                delete tmp;
            }
            else if (node->ch[]==null)
            {
                Node *tmp=node;
                node=node->ch[];
                delete tmp;
            }
            else
            {
                int dd=(node->ch[]->pri)<(node->ch[]->pri)?:;
                rotate(node,dd);
                remove(node->ch[dd^],x);
            }
        }
    }
    else remove(node->ch[d],x);
    node->push_up();
}

int rk(const Node* node,int x)
{
    if (node==null) return ;
    if (node->val<x) return node->cnt+node->ch[]->sz+rk(node->ch[],x);
    else return rk(node->ch[],x);
}

int kth(const Node* node,int k)
{
    if (node->ch[]==null)
    {
        if (node->cnt>=k) return node->val;
        else return kth(node->ch[],k-node->cnt);
    }
    else
    {
        if (node->ch[]->sz>=k) return kth(node->ch[],k);
        if (node->ch[]->sz+node->cnt>=k) return node->val;
        return kth(node->ch[],k-node->ch[]->sz-node->cnt);
    }
}

int pre(Node* node,int x)
{
    Node *t=node;
    int res=-0x3f3f3f3f;
    while (t!=null)
    {
        if (t->val>=x) t=t->ch[];
        else res=t->val,t=t->ch[];
    }
    return res;
}

int suc(Node* node,int x)
{
    Node *t=node;
    int res=0x3f3f3f3f;
    while (t!=null)
    {
        if (t->val<=x) t=t->ch[];
        else res=t->val,t=t->ch[];
    }
    return res;
}

int main()
{
    //srand((unsigned)time(NULL));
    null=new Node();
    null->cnt=;
    null->sz=;
    null->pri=-;
    null->ch[]=null;
    null->ch[]=null;
    tree=null;
    int n;
    scanf("%d",&n);
    while (n--)
    {
        int op,x;
        scanf("%d%d",&op,&x);
        switch(op)
        {
            case : insert(tree,x); break;
            case : remove(tree,x); break;
            case : printf("%d\n",rk(tree,x)+); break;
            case : printf("%d\n",kth(tree,x)); break;
            case : printf("%d\n",pre(tree,x)); break;
            case : printf("%d\n",suc(tree,x)); break;
        }
    }
    return ;
}