[题解]bzoj4825 HNOI2017单旋

Description

H 国是一个热爱写代码的国家,那里的人们很小去学校学习写各种各样的数据结构。伸展树(splay)是一种数据结构,因为代码好写,功能多,效率高,掌握这种数据结构成为了 H 国的必修技能。有一天,邪恶的“卡”带着他的邪恶的“常数”来企图毁灭 H 国。“卡”给 H 国的人洗脑说,splay 如果写成单旋的,将会更快。“卡”称“单旋 splay”为“spaly”。虽说他说的很没道理,但还是有 H 国的人相信了,小 H 就是其中之一,spaly 马上成为他的信仰。 而 H 国的国王,自然不允许这样的风气蔓延,国王构造了一组数据,数据由 m 个操作构成,他知道这样的数据肯定打垮 spaly,但是国王还有很多很多其他的事情要做,所以统计每个操作所需要的实际代价的任务就交给你啦。
数据中的操作分为五种:
1. 插入操作:向当前非空 spaly 中插入一个关键码为 key 的新孤立节点。插入方法为,先让 key 和根比较,如果 key 比根小,则往左子树走,否则往右子树走,如此反复,直到某个时刻,key 比当前子树根 x 小,而 x 的左子树为空,那就让 key 成为 x 的左孩子; 或者 key 比当前子树根 x 大,而 x 的右子树为空,那就让 key 成为 x 的右孩子。该操作的代价为:插入后,key 的深度。特别地,若树为空,则直接让新节点成为一个单个节点的树。(各节点关键码互不相等。对于“深度”的解释见末尾对 spaly 的描述)。
2. 单旋最小值:将 spaly 中关键码最小的元素 xmin 单旋到根。操作代价为:单旋前 xmin 的深度。(对于单旋操作的解释见末尾对 spaly 的描述)。
3. 单旋最大值:将 spaly 中关键码最大的元素 xmax 单旋到根。操作代价为:单旋前 xmax 的深度。
4. 单旋删除最小值:先执行 2 号操作,然后把根删除。由于 2 号操作之后,根没有左子树,所以直接切断根和右子树的联系即可(具体见样例解释)。 操作代价同 2 号操 作。
5. 单旋删除最大值:先执行 3 号操作,然后把根删除。 操作代价同 3 号操作。对于不是 H 国的人,你可能需要了解一些 spaly 的知识,才能完成国王的任务:
a. spaly 是一棵二叉树,满足对于任意一个节点 x,它如果有左孩子 lx,那么 lx 的关键码小于 x 的关键码。如果有右孩子 rx,那么 rx 的关键码大于 x 的关键码。
b. 一个节点在 spaly 的深度定义为:从根节点到该节点的路径上一共有多少个节点(包括自己)。
c. 单旋操作是对于一棵树上的节点 x 来说的。一开始,设 f 为 x 在树上的父亲。如果 x 为 f 的左孩子,那么执行 zig(x) 操作(如上图中,左边的树经过 zig(x) 变为了右边的树),否则执行 zag(x) 操作(在上图中,将
右边的树经过 zag(f) 就变成了左边的树)。每当执 行一次 zig(x) 或者 zag(x),x 的深度减小 1,如此反复,直到 x 为根。总之,单旋 x 就是通过反复执行 zig 和 zag 将 x 变为根。

Input

第一行单独一个正整数 m。
接下来 m 行,每行描述一个操作:首先是一个操作编号 c∈[1,5],即问题描述中给出的五种操作中的编号,若 c = 1,则再输入一个非负整数 key,表示新插入节点的关键码。1≤m≤10^5,1≤key≤10^9
所有出现的关键码互不相同。任何一个非插入操作,一定保证树非空。在未执行任何操作之前,树为空

Output

输出共 m 行,每行一个整数,第 i 行对应第 i 个输入的操作的代价。

Sample Input

5
1 2
1 1
1 3
4
5

Sample Output

1
2
2
2
2

Solution

单旋的时候对树的形态破坏不大,只会影响少数节点,所以用Splay维护深度就好了,自己手动分析一下影响,同时记录下在真正的Spaly树上的父亲、左右儿子。细节多如牛毛,真心难调。

代码:

#include<cstdio>
#include<algorithm>
using namespace std;

template<typename T>inline void read(T &x){
    T f=1;char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
    for(x=0;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
    x*=f;
}

const int maxn=100010,inf=0x7fffffff;
struct node{
    int ch[2],fa,key,dep,add,rfa,rch[2];
    node(){
        ch[0]=ch[1]=rch[0]=rch[1]=0;
        rfa=fa=key=dep=add=0;
    }
}T[maxn];
struct Splay_Tree{
    int root,num,rroot;
    Splay_Tree(){rroot=root=num=0;}
    int getson(int x){
        return x==T[T[x].fa].ch[1];
    }
    void pushdown(int x){
        if(!T[x].add)return;
        if(T[x].ch[0]){
            T[T[x].ch[0]].add+=T[x].add;
            T[T[x].ch[0]].dep+=T[x].add;
        }
        if(T[x].ch[1]){
            T[T[x].ch[1]].add+=T[x].add;
            T[T[x].ch[1]].dep+=T[x].add;
        }
        T[x].add=0;
    }
    void rotate(int x){
        int fa=T[x].fa,fafa=T[fa].fa,k=getson(x);
        pushdown(fa);pushdown(x);
        T[fa].ch[k]=T[x].ch[k^1];
        if(T[x].ch[k^1])T[T[x].ch[k^1]].fa=fa;
        T[x].ch[k^1]=fa;T[fa].fa=x;T[x].fa=fafa;
        if(fafa)T[fafa].ch[fa==T[fafa].ch[1]]=x;
    }
    void Splay(int x,int f=0){
        for(int fa;(fa=T[x].fa)!=f;rotate(x))
            if(T[fa].fa!=f)rotate(getson(x)==getson(fa)?fa:x);
        if(!f)root=x;
    }
    void Insert(int val,int dp){
        if(!root){
            T[root=++num].key=val;
            T[root].dep=dp;rroot=num;
            return;
        }
        int x=root,fa;
        while(x){
            pushdown(fa=x);
            x=T[x].ch[val>T[x].key];
        }
        T[fa].ch[val>T[fa].key]=++num;
        T[num].key=val;T[num].dep=dp;
        T[num].fa=fa;
        Splay(num);
    }
    int Find(int val){
        int x=root;
        while(1){
            pushdown(x);
            if(T[x].key==val)return x;
            x=T[x].ch[val>T[x].key];
        }
    }
    int Getpre(int val){
        int x=root,ans=-inf;
        while(x){
            pushdown(x);
            if(T[x].key<val)ans=x,x=T[x].ch[1];
            else x=T[x].ch[0];
        }
        return ans;
    }
    int Getsuf(int val){
        int x=root,ans=inf;
        while(x){
            pushdown(x);
            if(T[x].key>val)ans=x,x=T[x].ch[0];
            else x=T[x].ch[1];
        }
        return ans;
    }
    int Getmin(){
        int x=root;
        while(T[x].ch[0])pushdown(x),x=T[x].ch[0];
        return x;
    }
    int Getmax(){
        int x=root;
        while(T[x].ch[1])pushdown(x),x=T[x].ch[1];
        return x;
    }
}tree;
int n;

int main(){
    read(n);
    while(n--){
        int opt,x,ans;
        read(opt);
        if(opt==1){
            read(x);
            int l=tree.Getpre(x);
            int r=tree.Getsuf(x);
            if(l<0&&r==inf)tree.Insert(x,ans=1);
            else if(l<0){
                tree.Insert(x,ans=T[r].dep+1);
                T[T[r].rch[0]=tree.num].rfa=r;
            }
            else if(r==inf){
                tree.Insert(x,ans=T[l].dep+1);
                T[T[l].rch[1]=tree.num].rfa=l;
            }
            else{
                if(T[l].dep>T[r].dep){
                    tree.Insert(x,ans=T[l].dep+1);
                    T[T[l].rch[1]=tree.num].rfa=l;
                }
                else{
                    tree.Insert(x,ans=T[r].dep+1);
                    T[T[r].rch[0]=tree.num].rfa=r;
                }
            }
        }
        else if(opt==2){
            x=tree.Getmin();ans=T[x].dep;
            T[x].dep=1;tree.Splay(x);
            if(T[x].rfa){
                if(T[x].ch[1]){
                    T[T[x].ch[1]].add++;
                    T[T[x].ch[1]].dep++;
                }
                if(T[x].rch[1]){
                    int temp=T[x].rch[1];
                    T[temp].rfa=T[x].rfa;
                    T[T[x].rfa].rch[0]=temp;
                    tree.Splay(T[x].rfa,x);
                    T[T[T[x].rfa].ch[0]].add--;
                    T[T[T[x].rfa].ch[0]].dep--;
                }
                else T[T[x].rfa].rch[0]=0;
                T[x].rch[1]=tree.rroot;
                T[tree.rroot].rfa=x;
                T[x].rfa=T[x].rch[0]=0;
                tree.rroot=x;
            }
        }
        else if(opt==3){
            x=tree.Getmax();ans=T[x].dep;
            T[x].dep=1;tree.Splay(x);
            if(T[x].rfa){
                if(T[x].ch[0]){
                    T[T[x].ch[0]].add++;
                    T[T[x].ch[0]].dep++;
                }
                if(T[x].rch[0]){
                    int temp=T[x].rch[0];
                    T[temp].rfa=T[x].rfa;
                    T[T[x].rfa].rch[1]=temp;
                    tree.Splay(T[x].rfa,x);
                    T[T[T[x].rfa].ch[1]].add--;
                    T[T[T[x].rfa].ch[1]].dep--;
                }
                else T[T[x].rfa].rch[1]=0;
                T[x].rch[0]=tree.rroot;
                T[tree.rroot].rfa=x;
                T[x].rfa=T[x].rch[1]=0;
                tree.rroot=x;
            }
        }
        else if(opt==4){
            x=tree.Getmin();ans=T[x].dep;
            T[x].dep=1;tree.Splay(x);
            if(T[x].rfa){
                if(T[x].ch[1]){
                    T[T[x].ch[1]].add++;
                    T[T[x].ch[1]].dep++;
                }
                if(T[x].rch[1]){
                    int temp=T[x].rch[1];
                    T[temp].rfa=T[x].rfa;
                    T[T[x].rfa].rch[0]=temp;
                    tree.Splay(T[x].rfa,x);
                    T[T[T[x].rfa].ch[0]].add--;
                    T[T[T[x].rfa].ch[0]].dep--;
                }
                else T[T[x].rfa].rch[0]=0;
                T[x].rch[1]=tree.rroot;
                T[tree.rroot].rfa=x;
                T[x].rfa=T[x].rch[0]=0;
                tree.rroot=x;
            }
            tree.pushdown(tree.root);
            tree.rroot=T[tree.rroot].rch[1];
            T[tree.rroot].rfa=0;
            tree.root=T[tree.root].ch[1];
            T[tree.root].fa=0;
            if(tree.root){
                T[tree.root].add--;
                T[tree.root].dep--;
            }
        }
        else if(opt==5){
            x=tree.Getmax();ans=T[x].dep;
            T[x].dep=1;tree.Splay(x);
            if(T[x].rfa){
                if(T[x].ch[0]){
                    T[T[x].ch[0]].add++;
                    T[T[x].ch[0]].dep++;
                }
                if(T[x].rch[0]){
                    int temp=T[x].rch[0];
                    T[temp].rfa=T[x].rfa;
                    T[T[x].rfa].rch[1]=temp;
                    tree.Splay(T[x].rfa,x);
                    T[T[T[x].rfa].ch[1]].add--;
                    T[T[T[x].rfa].ch[1]].dep--;
                }
                else T[T[x].rfa].rch[1]=0;
                T[x].rch[0]=tree.rroot;
                T[tree.rroot].rfa=x;
                T[x].rfa=T[x].rch[1]=0;
                tree.rroot=x;
            }
            tree.pushdown(tree.root);
            tree.rroot=T[tree.rroot].rch[0];
            T[tree.rroot].rfa=0;
            tree.root=T[tree.root].ch[0];
            T[tree.root].fa=0;
            if(tree.root){
                T[tree.root].add--;
                T[tree.root].dep--;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值