AC日记——【模板】普通平衡树（Treap/SBT） 洛谷 P3369

【模板】普通平衡树（Treap/SBT）

思路：

　　劳资敲了一个多星期；

　　劳资终于a了；

　　劳资一直不a是因为一个小错误；

　　劳资最后看的模板；

　　劳资现在很愤怒；

　　劳资不想谈思路！！！

来，上代码：

#include <cstdio>
 
using namespace std;
 
#define maxn 1000005
 
struct SplayTreeNodeType {
    int w,key,opi,size,ch[];
};
struct SplayTreeNodeType tree[maxn];
 
int n,root,tot;
 
inline void updata(int now)
{
    tree[now].size=tree[now].w;
    if(tree[now].ch[]) tree[now].size+=tree[tree[now].ch[]].size;
    if(tree[now].ch[]) tree[now].size+=tree[tree[now].ch[]].size;
}
 
inline int pre()
{
    int now=tree[root].ch[];
    while(tree[now].ch[]) now=tree[now].ch[];
    return now;
}
 
inline int suc()
{
    int now=tree[root].ch[];
    while(tree[now].ch[]) now=tree[now].ch[];
    return now;
}
 
inline int getson(int now)
{
    return tree[tree[now].opi].ch[]==now;
}
 
void rotate(int now)
{
    int opi=tree[now].opi,fopi=tree[opi].opi,pos=getson(now);
    tree[opi].ch[pos]=tree[now].ch[pos^];
    tree[tree[opi].ch[pos]].opi=opi;
    tree[now].ch[pos^]=opi;tree[opi].opi=now;
    tree[now].opi=fopi;
    if(fopi) tree[fopi].ch[tree[fopi].ch[]==opi]=now;
    updata(opi),updata(now);
}
 
void splay(int now)
{
    for(int opi;opi=tree[now].opi;rotate(now))
    {
        if(tree[opi].opi) rotate(getson(now)==getson(opi)?opi:now);
    }
    root=now;
}
 
int rank(int x)
{
    int now=root,ans=;
    while()
    {
        if(x<tree[now].key) now=tree[now].ch[];
        else
        {
            ans+=tree[now].ch[]?tree[tree[now].ch[]].size:;
            if(x==tree[now].key)
            {
                splay(now);
                return ans+;
            }
            ans+=tree[now].w;
            now=tree[now].ch[];
        }
    }
}
 
int rank_(int x)
{
    int now=root;
    while()
    {
        if(tree[now].ch[]&&x<=tree[tree[now].ch[]].size) now=tree[now].ch[];
        else
        {
            int tmp=(tree[now].ch[]?tree[tree[now].ch[]].size:)+tree[now].w;
            if(x<=tmp) return tree[now].key;
            x-=tmp;now=tree[now].ch[];
        }
    }
}
 
inline void clear(int now)
{
    tree[now].ch[]=tree[now].ch[]=tree[now].w=tree[now].size=tree[now].key=tree[now].opi=;
}
 
inline void create(int x)
{
    tree[++tot].key=x;
    tree[tot].w=tree[tot].size=;
    tree[tot].ch[]=tree[tot].ch[]=tree[tot].opi=;
}
 
void insert(int x)
{
    if(!root) create(x),root=tot;
    else
    {
        int now=root,opi=;
        while()
        {
            if(tree[now].key==x)
            {
                tree[now].w++;
                tree[now].size++;
                splay(now);
                break;
            }
            opi=now;
            now=tree[now].ch[x>tree[opi].key];
            if(!now)
            {
                create(x);
                tree[tot].opi=opi;
                tree[opi].ch[x>tree[opi].key]=tot;
                splay(tot);
                break;
            }
        }
    }
}
 
void del(int x)
{
    int t=rank(x);
    if(tree[root].w>)
    {
        tree[root].w--;
        tree[root].size--;
        return ;
    }
    if(!tree[root].ch[]&&!tree[root].ch[])
    {
        clear(root);
        root=;
        return ;
    }
    if(!tree[root].ch[])
    {
        int tmp=root;
        root=tree[root].ch[];
        tree[root].opi=;
        clear(tmp);
        return ;
    }
    if(!tree[root].ch[])
    {
        int tmp=root;
        root=tree[root].ch[];
        tree[root].opi=;
        clear(tmp);
        return ;
    }
    int pre1=pre(),tmp=root;
    splay(pre1);
    tree[root].ch[]=tree[tmp].ch[];
    tree[tree[tmp].ch[]].opi=root;
    clear(tmp);updata(root);
}
 
inline void in(int &now)
{
    register int if_z=;now=;
    register char Cget=getchar();
    while(Cget>''||Cget<'')
    {
        if(Cget=='-') if_z=-;
        Cget=getchar();
    }
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
    now*=if_z;
}
 
int main()
{
    in(n);int ty,x;
    while(n--)
    {
        in(ty),in(x);
        if(ty==) insert(x);
        if(ty==) del(x);
        if(ty==) printf("%d\n",rank(x));
        if(ty==) printf("%d\n",rank_(x));
        if(ty==) insert(x),printf("%d\n",tree[pre()].key),del(x);
        if(ty==) insert(x),printf("%d\n",tree[suc()].key),del(x);
    }
    return ;
}