[那些你所不知道的鬼畜写法]平衡树学习笔记

treap:  (Orz fhq 大神，我曾经以为我会了 treap ，就再也不会写 splay 了，然后我遇上了 lct )

#include <cstdlib>
const int sizeOfMemory=10000;
template <class type> inline void swap(type & x, type & y) {type t=x; x=y; y=t;}

namespace treap
{
    struct node
    {
        int key, size;
        short weight;
        bool rev;
        node * left, * right;
        inline node():size(1), weight(0) {left=right=this;}
        inline void maintain() {size=left->size+right->size+1;}
        inline void pushdown() {if (rev) swap(left, right), left->rev^=1, right->rev^=1, rev=0;}
    };
    node * null=new node();
    node memory[sizeOfMemory], * portOfMemory=memory;

    inline node * newnode(int key)
    {
        node * ret=portOfMemory++;
        ret->key=key; ret->size=1;
        ret->weight=rand();
        ret->rev=0;
        ret->left=ret->right=null;
        return ret;
    }
    void splitBySize(node * t, int k, node *& l, node *& r)
    {
        if (t==null) l=r=null;
        else
        {
            t->pushdown();
            if (t->left->size<k)
            {
                l=t;
                splitBySize(t->right, k-t->left->size-1, t->right, r);
                l->maintain();
            }
            else
            {
                r=t;
                splitBySize(t->left, k, l, t->left);
                r->maintain();
            }
        }
    }
    void splitByKey(node * t, int k, node *& l, node *& r)
    {
        if (t==null) l=r=null;
        else
        {
            t->pushdown();
            if (t->key<k)
            {
                l=t;
                splitByKey(t->right, k, t->right, r);
                l->maintain();
            }
            else
            {
                r=t;
                splitByKey(t->left, k, l, t->left);
                r->maintain();
            }
        }
    }
    node * merge(node * l, node * r)
    {
        if (l==null) return r;
        if (r==null) return l;
        if (l->weight>r->weight)
        {
            l->pushdown();
            l->right=merge(l->right, r);
            l->maintain();
            return l;
        }
        else
        {
            r->pushdown();
            r->left=merge(l, r->left);
            r->maintain();
            return r;
        }
    }
}

splay:  (我曾经以为我会了 treap ，就再也不会在 lct 外的地方写 splay 了，然后我遇上了 自顶向下 splay )

等待完坑

AVL:  ( fhq 大神 说是为了解决人工实现 rope 时，treap 不能直接复制节点否则权重会痿掉的问题而搞出来的新产物，但 clj 大神似乎有更高明的做法)

等待完坑