BZOJ3223 文艺平衡树(splay)

题目背景

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

题目描述

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

输入输出格式

输入格式：
第一行为n,m n表示初始序列有n个数，这个序列依次是 (1,2, \cdots n-1,n)(1,2,⋯n−1,n) m表示翻转操作次数

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

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

输入输出样例

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

题解:打标记的splay!调了一个上午终于调出来了!继续挖坑~

代码如下:

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 100010
using namespace std;

struct Splay
{
    int key[N],son[N][],size[N],fa[N],tag[N],rt,sz,pos;
    inline void push_up(int x)
    {
        size[x]=size[son[x][]]+size[son[x][]]+;
    }
    inline void push_down(int x)
    {
        if(!tag[x])
        {
            return;
        }
        tag[x]=;
        tag[son[x][]]^=;
        tag[son[x][]]^=;
        swap(son[x][],son[x][]);
    }
    inline void rotate(int x)
    {
        int y=fa[x],z=fa[y],k=(son[y][]==x);
        son[z][son[z][]==y]=x;
        fa[x]=z;
        son[y][!k]=son[x][k];
        fa[son[x][k]]=y;
        son[x][k]=y;
        fa[y]=x;
        push_up(y);
    }
    inline void splay(int x,int goal)
    {
        for(push_down(x); fa[x]!=goal; rotate(x))
        {
            int y=fa[x],z=fa[y];
            if(z!=goal)
            {
                (son[y][]==x)^(son[z][]==y)?rotate(x):rotate(y);
            }
        }
        push_up(x);
        if(!goal)
        {
            rt=x;
        }
    }
    inline int kth(int k)
    {
        for(int x=rt;;)
        {
            push_down(x);
            int y=son[x][];
            if(k>size[y]+)
            {
                k-=size[y]+;
                x=son[x][];
            }
            else
            {
                if(k>size[y])
                {
                    return x;
                }
                else
                {
                    x=y;
                }
            }
        }
    }
    inline void reverse(int l,int r)
    {
        if(l>r)
        {
            swap(l,r);
        }
        int x=kth(l),y=kth(r+);
        splay(x,);
        splay(y,x);
        tag[son[y][]]^=;
    }
    inline int build(int *a,int father,int l,int r)
    {
        if(l>r)
        {
            return ;
        }
        int mid=(l+r)>>,x=++sz;
        fa[x]=father;
        key[x]=a[mid];
        son[x][]=build(a,x,l,mid-);
        son[x][]=build(a,x,mid+,r);
        push_up(x);
        return x;
    }
    inline void put(int x)
    {
        push_down(x);
        if(son[x][])
        {
            put(son[x][]);
        }
        if(key[x]>-1e9&&key[x]<1e9)
        {
            printf("%d ",key[x]);
        }
        if(son[x][])
        {
            put(son[x][]);
        }
    }
} splay1;

int main()
{
    int n,m,l,r,a[];

    scanf("%d%d",&n,&m);
    for(int i=; i<=n+; i++)
    {
        a[i]=i-;
    }
    a[]=-1e9;
    a[n+]=1e9;
    splay1.rt=splay1.build(a,,,n+);
    for(int i=; i<=m; i++)
    {
        scanf("%d%d",&l,&r);
        splay1.reverse(l,r);
    }
    splay1.put(splay1.rt);
}