洛谷 P3391 【模板】文艺平衡树(Splay)

题目背景

这是一道经典的Splay模板题——文艺平衡树。

题目描述

您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作:翻转一个区间,例如原有序序列是5 4 3 2 1,翻转区间是[2,4]的话,结果是5 2 3 4 1

输入输出格式

输入格式:

第一行为n,m n表示初始序列有n个数,这个序列依次是(1,2,⋯n−1,n)  m表示翻转操作次数

接下来m行每行两个数 [l,r]数据保证 1≤l≤r≤n

输出格式:

输出一行n个数字,表示原始序列经过m次变换后的结果

输入输出样例

输入样例#1: 
5 3
1 3
1 3
1 4
输出样例#1: 
4 3 2 1 5

说明

n,m≤100000 n, m

 

Splay的简介:

 

1 简介:
伸展树,或者叫自适应查找树,是一种用于保存有序集合的简单高效的数据结构。伸展树实质上是一个二叉查找树。允许查找,插入,删除,删除最小,删除最大,分割,合并等许多操作,这些操作的时间复杂度为O(logN)。由于伸展树可以适应需求序列,因此他们的性能在实际应用中更优秀。
伸展树支持所有的二叉树操作。伸展树不保证最坏情况下的时间复杂度为O(logN)。伸展树的时间复杂度边界是均摊的。尽管一个单独的操作可能很耗时,但对于一个任意的操作序列,时间复杂度可以保证为O(logN)。

 

2 自调整和均摊分析:
    平衡查找树的一些限制:
1、平衡查找树每个节点都需要保存额外的信息。
2、难于实现,因此插入和删除操作复杂度高,且是潜在的错误点。
3、对于简单的输入,性能并没有什么提高。
    平衡查找树可以考虑提高性能的地方:
1、平衡查找树在最差、平均和最坏情况下的时间复杂度在本质上是相同的。
2、对一个节点的访问,如果第二次访问的时间小于第一次访问,将是非常好的事情。
3、90-10法则。在实际情况中,90%的访问发生在10%的数据上。
4、处理好那90%的情况就很好了。

 

3 均摊时间边界:
在一颗二叉树中访问一个节点的时间复杂度是这个节点的深度。因此,我们可以重构树的结构,使得被经常访问的节点朝树根的方向移动。尽管这会引入额外的操作,但是经常被访问的节点被移动到了靠近根的位置,因此,对于这部分节点,我们可以很快的访问。根据上面的90-10法则,这样做可以提高性能。
为了达到上面的目的,我们需要使用一种策略──旋转到根(rotate-to-root)。

  上面所说的,我来举个例子解释一下:假设有这样一道题,有100000次操作,每次输入a,b,若a为0表示将b放入数列中,若a为1表示输出第b大的数。这道题看似简单,不就直接二叉搜索树嘛!但是如果数b是单调递增出现的,则树会成链,那么还是O(n^2)的复杂度。此时我们边用到了Splay,由于splay是不断翻转的,所以就算某一时刻他成了一条链,也会马上旋转而变成另外的形态(深度减低),通过这样不断地变换可以防止长期停留在链的状态,以保证每次操作平均复杂度O(log n)。

 

Splay的实现:

有话要说:

  关于Splay,我觉得自己已经完全掌握了,让我口头说还可以,但是要写篇详解实在是时间又少而且没精力(而且大神们的博客已经写的非常到位了,自己写的肯定不及他们),所以这里我提供本人自学Splay时所看的一些比较有用的博客:1、基础(非指针)    2、基础(指针)   3、应用

认真看上述博客并思考,便会发现Splay其实很简单。

 

Splay应用:

  Splay Tree可以方便的解决一些区间问题,根据不同形状二叉树先序遍历结果不变的特性,可以将区间按顺序建二叉查找树。 每次自下而上的一套splay都可以将x移动到根节点的位置,利用这个特性,可以方便的利用Lazy的思想进行区间操作。 对于每个节点记录size,代表子树中节点的数目,这样就可以很方便地查找区间中的第k小或第k大元素。 对于一段要处理的区间[x, y],首先splay x-1到root,再splay y+1到root的右孩子,这时root的右孩子的左孩子对应子树就是整个区间。   这样,大部分区间问题都可以很方便的解决,操作同样也适用于一个或多个条目的添加或删除,和区间的移动。   参考例题:bzoj3224   操作:1. 插入x数
     2. 删除x数(若有多个相同的数,因只删除一个)
     3. 查询x数的排名(若有多个相同的数,因输出最小的排名)
     4. 查询排名为x的数
     5. 求x的前驱(前驱定义为小于x,且最大的数)
     6. 求x的后继(后继定义为大于x,且最小的数)  

关于这道题:

  只要我们弄懂Splay,其实本题很简单:首先按照中序遍历建树,然后对于每次修改区间l,r,首先得提出这段区间,方法是将l的前趋l-1旋转到根节点,将r的后趋r+1旋转到根节点的右儿子,我们可以自己画图试试,容易发现经过这个操作后,根节点的右儿子的左子树(具体应该说是这个左子树的中序遍历)就是区间l-r。关键的翻转时,因为树是中序遍历(左根右),所以我们只要将l-r(前面所说的根节点的右儿子的左子树)这个区间子树左右儿子的节点交换位置(这样再中序遍历相当于右根左,即做到了翻转操作)。关键是翻转的优化,我们用到懒惰标记lazy[x](表示x是否翻转),每次翻转时只要某个节点有标记且在翻转的区间内,则将标记下放给它的两个儿子节点且将自身标记清0,这样便避免了多余的重复翻转。(不懂画图看博客)

1、裸代码:

 

// luogu-judger-enable-o2
#include<bits/stdc++.h>
#define ll long long
#define il inline
#define debug printf("%d %s\n",__LINE__,__FUNCTION__)
using namespace std;
const int N=100005;
il int gi()
{
    int a=0;char x=getchar();bool f=0;
    while((x<'0'||x>'9')&&x!='-')x=getchar();
    if(x=='-')x=getchar(),f=1;
    while(x>='0'&&x<='9')a=a*10+x-48,x=getchar();
    return f?-a:a;
}
int n,m,tot,root,siz[N],fa[N],flag[N],key[N],ch[N][2],cnt[N],ans[N];
il void update(int rt)
{
    int l=ch[rt][0],r=ch[rt][1];
    siz[rt]=siz[l]+siz[r]+1;
}
il void pushdown(int now)
{
    if(flag[now]){
        flag[ch[now][0]]^=1;
        flag[ch[now][1]]^=1;
        swap(ch[now][0],ch[now][1]);
        flag[now]=0;
    }
}
il int getson(int x){return ch[fa[x]][1]==x;}
il void rotate(int x)
{
    int y=fa[x],z=fa[y],b=getson(x),c=getson(y),a=ch[x][!b];
    if(z)ch[z][c]=x;else root=x;fa[x]=z;
    if(a)fa[a]=y;ch[y][b]=a;
    ch[x][!b]=y;fa[y]=x;
    update(y);update(x);
}
il void splay(int x,int i)
{
    while(fa[x]!=i){
        int y=fa[x],z=fa[y];
        if(z==i)rotate(x);
        else {
            if(getson(x)==getson(y)){rotate(y);rotate(x);}
            else {rotate(x);rotate(x);}
        }
    }
}
il int find(int x)
{
    int now=root;
    while(1){
        pushdown(now);
        if(ch[now][0]&&x<=siz[ch[now][0]])now=ch[now][0];
        else {
            int tmp=(ch[now][0]?siz[ch[now][0]]:0)+1;
            if(x<=tmp)return now;
            x-=tmp;
            now=ch[now][1];
        }
    }
}
il int build(int l,int r,int rt)
{
    int now=l+r>>1;
    fa[now]=rt;
    key[now]=ans[now];
    if(l<now)ch[now][0]=build(l,now-1,now);
    if(r>now)ch[now][1]=build(now+1,r,now);
    update(now);
    return now;
}
il void print(int now)
{
    pushdown(now);
    if(ch[now][0])print(ch[now][0]);
    ans[++tot]=key[now];
    if(ch[now][1])print(ch[now][1]);
}
int main()
{
    n=gi(),m=gi();int x,y;
    for(int i=1;i<=n+2;i++)ans[i]=i-1;
    root=build(1,n+2,0);
    for(int i=1;i<=m;i++){
        x=gi(),y=gi();
        x=find(x),y=find(y+2);
        splay(x,0);splay(y,x);
        flag[ch[ch[root][1]][0]]^=1;
    }
    print(root);
    for(int i=1;i<=n;i++)printf("%d ",ans[i+1]);
    return 0;
}

 

 

 

 

 2、方便理解,带注释代码:

 

/*Splay只记模板是很困难的,而且真正运用时易生疏出错,所以必须理解,在看代码前先弄懂
Splay的原理,这篇代码是带注释的Splay模板,题目来自洛谷P3391 ———————————by 520*/
#include<bits/stdc++.h>
#define il inline
#define debug printf("%d %s\n",__LINE__,__FUNCTION__)
using namespace std;
const int N=100005;
il int gi()
{
    int a=0;char x=getchar();bool f=0;
    while((x<'0'||x>'9')&&x!='-')x=getchar();
    if(x=='-')x=getchar(),f=1;
    while(x>='0'&&x<='9')a=a*10+x-48,x=getchar();
    return f?-a:a;
}

int n,m,tot,root,siz[N],fa[N],lazy[N],key[N],tree[N][2],ans[N];
/*root为根节点,siz存储子树节点数,fa存储父节点,lazy是懒惰标记用来标记区间翻转操作,key数组存储原数列,tree为
splay树,ans存储答案*/

il void pushup(int rt)  //作用类似与线段树
{
    int l=tree[rt][0],r=tree[rt][1];         //pushup作用是将子树的节点个数更新给根节点
    siz[rt]=siz[l]+siz[r]+1;
}

il void pushdown(int now)
{
    if(lazy[now]){
        lazy[tree[now][0]]^=1;
        lazy[tree[now][1]]^=1;               /*pushdown作用是下放懒惰标记,若某一节点所在子树(即某一区间)被翻转
        ,则将懒惰标记下放给两个儿子节点,交换左右儿子位置(中序遍历,交换左右儿子后相当于翻转)并对所在子树根节
        点的标记清0,*/
        swap(tree[now][0],tree[now][1]);
        lazy[now]=0;
    }
}

il int getson(int x){return tree[fa[x]][1]==x;}  //getson判断x是其父亲的右儿子还是左儿子

il void rotate(int x)       //旋转操作,直接写在一个函数里,可以称为上旋
{
    int y=fa[x],z=fa[y],b=getson(x),c=getson(y),a=tree[x][!b];  /*y是x的父节点,z是y的父节点,getson解释过了。
        特别解释一下a,a为旋转时需要移动的子树,若x为左儿子则右旋时要将x的右子树移动,同理若x为右儿子则左旋时要
        将x的左子树移动,所以这里a=tree[x][!b]*/
    if(z)tree[z][c]=x;else root=x;fa[x]=z; /*若z不为根节点,则用x替代y的位置;若z为根节点,则将x变为根节点。*/
    if(a)fa[a]=y;tree[y][b]=a; /*若存在要移动的子树a,则把a和y相连,取代原来x的位置*/
    tree[x][!b]=y;fa[y]=x;  /*!b的原因:若x为左儿子则旋转后y为x的右儿子,若x为右儿子则旋转后y为x的左儿子。记得将y
                            指向x*/
    pushup(y);pushup(x);   /*旋转后,对被移动了的y和x更新它们各自的子树节点数*/
}

il void splay(int x,int i)
{
    while(fa[x]!=i){          //只要x没有旋转到需要的点下面,则一直旋,注意根节点的父亲为虚点0
        int y=fa[x],z=fa[y];
        if(z==i)rotate(x);     //若x的爷爷是i,则只需旋一次
        else {    
            if(getson(x)==getson(y)){rotate(y);rotate(x);}   /*若x和y为相同偏向,则进行Zig-Zig或Zag-Zag操作*/
            else {rotate(x);rotate(x);}   /*否则进行Zig-Zag或Zag-Zig操作*/
                /*注意rotate函数中已经包含了这四种操作情况了*/
        }
    }
}

il int find(int x)    //查找x的位置
{
    int now=root;    //从根节点往下
    while(1){
        pushdown(now);    //本次操作要将前面的标记进行翻转
        if(tree[now][0]&&x<=siz[tree[now][0]])now=tree[now][0];   //若存在左子树且x小于等于左子树的节点数,则x在左子树上
        else {
            int tmp=(tree[now][0]?siz[tree[now][0]]:0)+1;   //往右子树找,+1代表加上这个子树的根节点
            if(x==tmp)return now;      //若找到了x,返回它的位置
            x-=tmp;      //否则x减去根节点右子树以外的节点数,这个画图能理解,因为siz值并不是直接的x的值
            now=tree[now][1];  //将原来根节点的右儿子赋为新的根节点,继续递归查找x位置
        }
    }
}

il int build(int l,int r,int rt)   //建树过程和线段树类似
{
    int now=l+r>>1;
    fa[now]=rt;
    key[now]=ans[now];        //key存原数组1到n,准确说是0到n+1,原因是主函数里的预处理
    if(l<now)tree[now][0]=build(l,now-1,now);
    if(r>now)tree[now][1]=build(now+1,r,now);
    pushup(now);   //记得pushup
    return now;
}

il void print(int now)   //输出时中序遍历,按左根右输出
{
    pushdown(now);   //记得要翻转
    if(tree[now][0])print(tree[now][0]);   //因为中序遍历左根右,所以递归根节点左子树到第一个数的位置
    ans[++tot]=key[now];   //回溯时存储答案,注意我们翻转操作的是原数组下标
    if(tree[now][1])print(tree[now][1]);   //同理递归根节点的右子树
}

int main()
{
    n=gi(),m=gi();int x,y;
    for(int i=1;i<=n+2;i++)ans[i]=i-1;    /*因为取出操作区间时旋转的是x的前驱和y的后驱,所以预处理时第i个点
        存的是i的前驱*/
    root=build(1,n+2,0);
    while(m--)
    {
        x=gi(),y=gi();
        x=find(x),y=find(y+2);  /*查找x的前驱所在的位置,和y后驱所在的位置,因为预处理时ans存的是前趋,
                                所以直接查找x,而y的后驱变成了y+2*/
        splay(x,0);splay(y,x);  /*将x前驱上旋至根节点,y的后驱上旋成根节点右儿子的左子树*/
        lazy[tree[tree[root][1]][0]]^=1;//经过旋转后,此时根节点的右儿子的左子树就是需要翻转的区间,所以lazy标记
    }
    print(root);
    for(int i=1;i<=n;i++)printf("%d ",ans[i+1]);   //输出时将前驱还原为原数
    return 0;
}

 

 

 

 

上一篇:luogu P3391 【模板】文艺平衡树(Splay)


下一篇:P3391 【模板】文艺平衡树(Splay) 平衡树