BZOJ 3223: Tyvj 1729 文艺平衡树

Description

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

Input

第一行为n,m n表示初始序列有n个数，这个序列依次是(1,2……n-1,n) m表示翻转操作次数
接下来m行每行两个数[l,r] 数据保证 1<=l<=r<=n

Output

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

Sample Input

5 3

1 3

1 3

1 4

Sample Output

4 3 2 1 5

HINT

N,M<=100000

Source

平衡树

Solution

• 好久没打平衡树，就打了个 Splay 练了练以前不会的翻转操作。

• 比如说要翻转一个区间 $[l,r]$ ，就把 $l-1$ 旋到根，再把 $r+1$ 旋到 $l-1$ 的右儿子。

• 这样区间 $[l,r]$ 就是 $r+1$ 的左儿子了，打上（异或）翻转标记即可（记得下传）。

• 注意所说的 $l$ 、$r$ 不是编号 $l,r$ ，而是下标，需要查询所在位置（查询第 $k$ 小操作）。

• 输出时直接输出中序遍历即可。

• 时间复杂度 $O(N\ log\ N)$ 。

Code

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e5+3,inf=1e9;
int root,tot;
int a[N],fa[N],key[N],s[N][2],size[N];
bool bz[N];
inline int read()
{
    int X=0,w=1; char ch=0;
    while(ch<'0' || ch>'9') {if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();
    return X*w;
}
inline void write(int x)
{
    if(x>9) write(x/10);
    putchar(x%10+'0');
}
inline bool pd(int x)
{
    return x==s[fa[x]][1];
}
inline void update(int x)
{
    if(x) size[x]=size[s[x][0]]+size[s[x][1]]+1;
}
inline void down(int x)
{
    if(!bz[x]) return;
    bz[s[x][0]]^=1,bz[s[x][1]]^=1;
    swap(s[x][0],s[x][1]);
    bz[x]=0;
}
inline void build(int &v,int l,int r)
{
    if(l>r) return;
    int mid=(l+r)>>1;
    key[v=++tot]=a[mid],size[v]=1;
    build(s[v][0],l,mid-1);
    build(s[v][1],mid+1,r);
    update(fa[s[v][0]]=fa[s[v][1]]=v);
}
inline void rotate(int x)
{
    down(x);
    int y=fa[x],w=pd(x);
    if(s[y][w]=s[x][w^1]) fa[s[y][w]]=y;
    if(fa[x]=fa[y]) s[fa[y]][pd(y)]=x;
    update(s[fa[y]=x][w^1]=y);
}
inline void splay(int x,int k)
{
    for(int y;(y=fa[x])!=k;rotate(x))
        if(fa[y]!=k) rotate(pd(x)==pd(y)?y:x);
    update(x);
    if(!k) root=x;
}
inline int kth(int x,int k)
{
    down(x);
    if(size[s[x][0]]+1==k) return x;
    if(k<=size[s[x][0]]) return kth(s[x][0],k);
    return kth(s[x][1],k-size[s[x][0]]-1);
}
inline void flip(int l,int r)
{
    int x=kth(root,l-1),y=kth(root,r+1);
    splay(x,0),splay(y,x);
    bz[s[y][0]]^=1;
}
inline void print(int v)
{
    down(v);
    if(s[v][0]) print(s[v][0]);
    if(key[v]<inf) write(key[v]),putchar(' ');
    if(s[v][1]) print(s[v][1]);
}
int main()
{
    int n=read(),m=read();
    a[1]=a[n+2]=inf;
    for(int i=2;i<=n+1;i++) a[i]=i-1;
    build(root,1,n+2);
    while(m--)
    {
        int l=read()+1,r=read()+1;
        flip(l,r);
    }
    print(root);
    return 0;
}