Splay

#include <iostream>
#include <ctime>
#include <cstdio>
#include <cctype>
const int maxn = 1e5+5;
struct Node
{
    int l,r;
    int val,size;
    int cnt;        //当前结点重复次数,默认为1
}spl[maxn];         //内存池
int cnt,root;       //内存池计数器与根节点编号
inline void newnode(int &now,int &val)
{
    spl[now=++cnt].val=val;
    spl[cnt].size++;
    spl[cnt].cnt++;
}
inline void update(int now) //更新size
{
    spl[now].size=spl[spl[now].l].size+spl[spl[now].r].size+spl[now].cnt;
}
inline void zig(int &now)
{
    int l = spl[now].l;
    spl[now].l = spl[l].r;
    spl[l].r = now;
    now = l;
    update(spl[now].r),update(now);
}
inline void zag(int &now)
{
    int r = spl[now].r;
    spl[now].r = spl[r].l;
    spl[r].l = now;
    now = r;
    update(spl[now].l),update(now);
}
void splaying(int x,int &y) //我要把x伸展到y那个位置!
{
    if(x==y) return;        //如果到了终点,return
    int &l = spl[y].l, &r = spl[y].r;   //temp
    if(x==l) zig(y);        //如果左儿子是终点,那就单旋
    else if(x==r) zag(y);   //右儿子是终点也是单旋
    else        //否则就一定是双旋
    {
        if(spl[x].val<spl[y].val)
        {
            if(spl[x].val<spl[l].val)
                splaying(x,spl[l].l),zig(y),zig(y);     //zigzig情况
            else splaying(x,spl[l].r),zag(l),zig(y);    //zagzig情况
        }
        else
        {
            if(spl[x].val>spl[r].val)
                splaying(x,spl[r].r),zag(y),zag(y);     //zagzag情况
            else splaying(x,spl[r].l),zig(r),zag(y);    //zigzag情况
        }
    }
}
inline void delnode(int now)
{
    splaying(now,root);     //将要删除的结点伸展至根结点
    if(spl[now].cnt>1) spl[now].size--,spl[now].cnt--;  //如果有重复,令重复次数--
    else if(spl[root].r)    //否则如果当前结点有后继
    {
        int p = spl[root].r;
        while(spl[p].l) p=spl[p].l;     //找到后继
        splaying(p,spl[root].r);        //将其伸展至根结点右儿子
        spl[spl[root].r].l=spl[root].l; //右儿子左儿子变为根结点
        root=spl[root].r;               //根结点变为根结点右儿子
        update(root);                   //更新一下size信息
    }
    else root = spl[root].l;    //伸展之后没有后继,说明它是最大的了,那就直接删除
}
void ins(int &now,int &val)
{
    if(!now) newnode(now,val),splaying(now,root);
    else if(val<spl[now].val) ins(spl[now].l,val);
    else if(val>spl[now].val) ins(spl[now].r,val);
    else spl[now].size++,spl[now].cnt++,splaying(now,root); //特判相同的情况
}
void del(int now,int &val)
{
    if(spl[now].val==val) delnode(now);
    else if(val<spl[now].val) del(spl[now].l,val);
    else del(spl[now].r,val);
}
//以下大致与以前的代码相同,有大变动的地方给出了注释
int getrank(int val)
{
    int now = root, rank = 1;
    while(now)
    {
        if(spl[now].val==val)   //找到了要的结点,这个之前的没有
        {
            rank+=spl[spl[now].l].size;
            splaying(now,root);
            break;
        }
        if(val<=spl[now].val)
            now=spl[now].l;
        else
        {
            rank+=spl[spl[now].l].size+spl[now].cnt;
            now=spl[now].r;
        }
    }
    return rank;
}
int getnum(int rank)
{
    int now = root;
    while(now)
    {
        int lsize = spl[spl[now].l].size;
        if(lsize+1<=rank&&rank<=lsize+spl[now].cnt) //如果在这个范围内,那就是当前结点
        {
            splaying(now,root);
            break;
        }
        else if(lsize>=rank)
            now=spl[now].l;
        else
        {
            rank-=lsize+spl[now].cnt;
            now=spl[now].r;
        }
    }
    return spl[now].val;
}
int main()
{
//    #ifndef ONLINE_JUDGE
//        freopen("in.in", "r", stdin);
//        freopen("out.out", "w", stdout);
//    #endif
//    clock_t c1 = clock();
    //======================================
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int opt,x;
        scanf("%d%d",&opt,&x);
        switch(opt)
        {
        case 1:
            ins(root,x);
            break;
        case 2:
            del(root,x);
            break;
        case 3:
            printf("%d\n",getrank(x));
            break;
        case 4:
            printf("%d\n",getnum(x));
            break;
        case 5:
            printf("%d\n",getnum(getrank(x)-1));
            break;
        case 6:
            printf("%d\n",getnum(getrank(x+1)));
            break;
        }
    }
    //std::cerr << "Time:" << clock() - c1 << "ms" << std::endl;
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值