Treap kth rank 模版

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>

using namespace std;

const int maxn = 3e4 + 100;

struct Node
{
    Node * ch[2];
    int r, v, s;
    Node (int v = 0): v(v)
    {
        ch[0] = ch[1] = NULL;
        r = rand();
        s = 1;
    }
    bool operator<(const Node & rhs) const
    {
        return r < rhs.r;
    }
    int cmp(int x)
    {
        if (x == v) return -1;
        return x < v ? 0 : 1;
    }
    void maintain()
    {
        s = 1;
        if (ch[0] != NULL) s += ch[0]->s;
        if (ch[1] != NULL) s += ch[1]->s;
    }
};

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

int find(Node *o, int x)
{
    while(o != NULL)
    {
        int d = o->cmp(x);
        if(d == -1) return 1;
        else o = o->ch[d];
    }
    return 0;
}

void insert(Node * (&o), int x)
{
    if (o == NULL) o = new Node(x);
    else
    {
        int d = (x < o->v ? 0 : 1);
        insert(o->ch[d], x);
        if (o->ch[d]->r > o->r) rotate(o, d ^ 1);
    }
    o->maintain();
}

void remove(Node * (&o), int x)
{
    int d = o->cmp(x);
    if (d == -1)
    {
        Node * u = o;
        if (o->ch[0] != NULL && o->ch[1] != NULL)
        {
            int d2 = (o->ch[0]->r > o->ch[1]->r) ? 1 : 0;
            rotate(o, d2);
            remove(o->ch[d2], x);
        }
        else
        {
            if (o->ch[0] == NULL) o = o->ch[1];
            else o = o->ch[0];
            delete u;
        }
    }
    else remove(o->ch[d], x);
    if (o != NULL) o->maintain();
}

int kth(Node * o, int k)
{
    if (o == NULL || k <= 0 || k > o->s) return 0;
    int s = (o->ch[1] == NULL ? 0 : o->ch[1]->s);
    if (k == s + 1) return o->v;
    else if (k <= s) return kth(o->ch[1], k);
    else return kth(o->ch[0], k - s - 1);
}

int rk(Node * o, int x)
{
    if (o == NULL) return 0;
    int s = (o->ch[1] == NULL ? 0 : o->ch[1]->s);
    if(x < o->v) return s + 1 + rk(o->ch[0], x);
    else return rk(o->ch[1], x);
}

void mergeto(Node * (&src), Node * (&dest))
{
    if (src->ch[0] != NULL) mergeto(src->ch[0], dest);
    if (src->ch[1] != NULL) mergeto(src->ch[1], dest);
    insert(dest, src->v);
    delete src;
    src = NULL;
}

void removetree(Node * (&x))
{
    if (x->ch[0] != NULL) removetree(x->ch[0]);
    if (x->ch[1] != NULL) removetree(x->ch[1]);
    delete x;
    x = NULL;
}

int n, m, a[maxn], b[maxn];
int main()
{
    Node * root = NULL;
    int q;
    char op[10];
    int x;

    for(scanf("%d", &q); q; --q)
    {
        scanf("%s%d", op, &x);
        if(op[0] == 'I')
        {
            if(find(root, -x) == 0) insert(root, -x);
        }
        else if(op[0] == 'D')
        {
            if(find(root, -x) == 1) remove(root, -x);
        }
        else if(op[0] == 'K')
        {
            if(root != NULL && x <= root->s) printf("%d\n", -kth(root, x));
            else puts("invalid");
        }
        else 
        {
            printf("%d\n", rk(root, -x));
        }
    }

    return 0;
}

// Input
// 8
// I -1
// I -1
// I 2
// C 0
// K 2
// D -1
// K 1
// K 2

// Output
// 1
// 2
// 2
// invalid