treap模版代码

treap模版暂存。

以后修改整理。

#include<cstdio>
#include<iostream>
#include <time.h>
#include<cstdlib>

using namespace std;
struct Node
{
    Node *ch[];//左右子树
    int r;//优先级。数值越大，优先级越高
    int v;//值
    int cmp(int x)const
    {
        ;
        :;
    }
};
/*
旋转操作：
1.取出新的根节点。左旋：取原根的右孩子；右旋：取原根的左孩子
2.左旋：原根的右孩子的左孩子取代之；右旋：原根的左孩子的右孩子取代之
3.左旋：原根成为新根的左孩子；右旋：原根成为新根的右孩子
4.将更新返回的根节点
旋转左右是根据原根结点的位置变化方向而定的
*/
void rotate(Node* &o,int d)//d=0代表左旋，d=1代表右旋
{
    Node *k=o->ch[d^];//左旋时d^1得到1右孩子，右旋时d^1得到0左孩子，取出该结点
    o->ch[d^]=k->ch[d]; //孩子的左（右）孩子取代孩子位置
    k->ch[d]=o;//原根结点成为右（左）结点的左（右）孩子
    o=k;//孩子结点取代原根节点位置，保证返回值是新的根结点
}

void insert(Node* &o,int x)//在以o为根的子树中插入键值x，修改o
{
    if(o==NULL)
    {
        o=new Node();
        o->ch[]=o->ch[]=NULL;
        o->v=x;
        o->r=rand();
    }
    else
    {
        int d=o->cmp(x);
        insert(o->ch[d],x);
        );
    }
}
void remove(Node* &o,int x)
{
    int d=o->cmp(x);
    )
    {
        ]==NULL) o=o->ch[];
        ]==NULL) o=o->ch[];
        else
        {
            ]->r > o->ch[]->r)?:;
            rotate(o,d2);
            remove(o->ch[d2],x);
        }
    }
    else remove(o->ch[d],x);
}
bool find(Node* o,int x)
{
    while(o!=NULL)
    {
        int d=o->cmp(x);
        ) return true;
        else o=o->ch[d];
    }
    return false;
}
Node *head;

int main()
{
    int n;
    scanf("%d",&n);
    ; i<=n; ++i)
    {
        int t;
        scanf("%d",&t);
        insert(head,t);
    }
    int p;
    cout<<"===="<<endl;
    while(scanf("%d",&p)!=EOF)
    {
        int t;
        scanf("%d",&t);
        switch(p)
        {
        :
            cout<<find(head,t)<<endl;
            break;
        :
            insert(head,t);
            break;
        :
            remove(head,t);
            break;

        }
    }
    ;
}