UESTC 395 Dynamic Query System --Treap

题意：让你维护一个集合，有8种操作：

1.  I x  插入一个数

2.  R x 删除x

3.  S 输出总的数个数（集合大小）

4.  L x  查询小于x的数的个数

5.  W k  查询集合中数从小到大排列的第k个数

6.  C x  查询x的个数

7.  MI  查询集合中最小的数

8.  MA 查询集合中最大的数

一道平衡树模拟的裸题，直接套版然后改下细节就行了。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;

struct node
{
    node *ch[];
    int v,r,sz,cnt;
    void UP(){ sz = ch[]->sz + ch[]->sz + cnt; }
    int cmp(int x)const
    {
        if(x == v) return -;
        return x > v;
    }
}*null,*root;

void create(node *&o,int v=)
{
    o = new node();
    o->sz = o->cnt = ;
    o->v = v;
    o->r = rand();
    o->ch[] = o->ch[] = null;
}

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

int countx(node *o,int v)
{
    while(o != null)
    {
        int d = o->cmp(v);
        if(d == -)
            return o->cnt;
        o = o->ch[d];
    }
    return ;
}

void insert(node *&o,int v)
{
    if(o == null)
    {
        create(o,v);
        return;
    }
    int d = o->cmp(v);
    if(d == -)
    {
        o->cnt++;
        o->UP();
        return;
    }
    insert(o->ch[d],v);
    if(o->r > o->ch[d]->r)
        rotate(o,d);
    if(o != null)
        o->UP();
}

void del(node *&o,int v)
{
    if(o == null) return;
    int d = o->cmp(v);
    if(d == -)
    {
        if(o->cnt > )
            o->cnt--;
        else
        {
            if(o->ch[] == null) o = o->ch[];
            else if(o->ch[] == null) o = o->ch[];
            else
            {
                int d2 = o->ch[]->r < o->ch[]->r;
                if(o->ch[d2] == null)
                {
                    delete o;
                    o = null;
                    return;
                }
                rotate(o,d2);
                del(o->ch[d2^],v);
            }
        }
    }
    else del(o->ch[d],v);
    if(o != null)
        o->UP();
}

int Kth(node *o,int k)
{
    if(o == null || k ==  || k > o->sz) return -;
    int left = o->ch[]->sz;
    if(k > left && k <= left + o->cnt)
        return o->v;
    if(k > left + o->cnt)
        return Kth(o->ch[],k-left-o->cnt);
    return Kth(o->ch[],k);
}

int Count(node *o,int v)
{
    if(o == null) return ;
    int left = o->ch[]->sz;
    if(o->v == v) return left;
    else if(v < o->v)
        return Count(o->ch[],v);
    return Count(o->ch[],v)+left+o->cnt;
}

void init()
{
    create(null);
    null->sz = ;
    root = null;
}

int main()
{
    int t,n,i,x;
    char ss[];
    scanf("%d",&t);
    while(t--)
    {
        init();
        scanf("%d",&n);
        while(n--)
        {
            scanf("%s",ss);
            if(ss[] == 'M')
            {
                if(ss[] == 'I')
                    printf("%d\n",Kth(root,));
                else
                    printf("%d\n",Kth(root,root->sz));
            }
            else if(ss[] == 'I')
            {
                scanf("%d",&x);
                insert(root,x);
            }
            else if(ss[] == 'S')
            {
                printf("%d\n",root->sz);
            }
            else if(ss[] == 'R')
            {
                scanf("%d",&x);
                del(root,x);
            }
            else if(ss[] == 'L')
            {
                scanf("%d",&x);
                printf("%d\n",Count(root,x));
            }
            else if(ss[] == 'W')
            {
                scanf("%d",&x);
                printf("%d\n",Kth(root,x));
            }
            else if(ss[] == 'C')
            {
                scanf("%d",&x);
                printf("%d\n",countx(root,x));
            }
        }
    }
    return ;
}