洛谷 P3391 模板Splay

#include<bits/stdc++.h>
using namespace std;
#define maxn 200000
int read()
{
    char ch=getchar();int f=,w=;
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(ch<=''&&ch>=''){w=w*+ch-'';ch=getchar();}
    return f*w;
}

int n,root,m,tot;

struct sj
{
    int ch[];            //左儿子和右儿子
    int ff,v;            //ff是父亲 v
    int size;            //size是儿子节点个数
    int mark;            //打上的标记
    void init(int x,int fa)                                    //初始化一个节点 左儿子和右儿子以及父亲都清零
        {
            ff=ch[]=ch[]=;
            size=;v=x;ff=fa;                                //含有子节点个数为1 父亲节点为参数fa
        }
}t[maxn];

inline void pushup(int x)
{
    t[x].size=t[t[x].ch[]].size+t[t[x].ch[]].size+;
}

inline void pushdown(int x)
{
    if(t[x].mark)
    {
        t[t[x].ch[]].mark^=;
        t[t[x].ch[]].mark^=;
        t[x].mark=;
        swap(t[x].ch[],t[x].ch[]);
    }
}

inline void rotate(int x)                //旋转函数
{
    int y=t[x].ff;
    int z=t[y].ff;                        //这是爷爷节点
    int k=t[y].ch[]==x;                //看x是左儿子还是右儿子 k取0,1
    t[z].ch[t[z].ch[]==y]=x;            //这是把爷爷节点的左/右儿子变成x
    t[x].ff=z;
    t[y].ch[k]=t[x].ch[k^];
    t[t[x].ch[k^]].ff=y;
    t[x].ch[k^]=y;
    t[y].ff=x;                            //此段画图理解
    pushup(y);pushup(x);                //再继续进行修改 push_up();
}

inline void splay(int x,int goal)
{
    while(t[x].ff!=goal)                //当我的这个节点还没有达到目标位置
    {
        int y=t[x].ff;int z=t[y].ff;
        if(z!=goal)                        //如果还没有跳到目标节点 继续旋转
        (t[z].ch[]==y)^(t[y].ch[]==x)?rotate(x):rotate(y);        //Warning:: !!!!同边现象时要先翻爸爸再翻儿子 否则不满足搜索树的性质!!!!
        rotate(x);
    }
    if(goal==)root=x;                    //如果目标节点是0 那么就把root 根节点变成x
}

inline void insert(int x)
{
    int u=root,ff=;                    //从根节点开始插入 一开始父亲变成0
    while(t[u].size!=)ff=u,u=t[u].ch[x>t[u].v];    //有根的时候 ff父亲变成根节点 u再继续下去向下操作
    u=++tot;                            //tot是实时增加的一个记录子节点个数的全局变量
    if(ff)t[ff].ch[x>t[ff].v]=u;        //如果父亲不为零 他的父亲的左/右儿子就是u
    t[u].init(x,ff);                    //初始化这个节点 并且它的父亲就是ff
    splay(u,);                            //把这个点旋转到根节点
}

inline int get_k(int k)
{
    int u=root;
    while()
    {
        pushdown(u);
        if(t[t[u].ch[]].size>=k)u=t[u].ch[];
        else if(t[t[u].ch[]].size+==k)return u;
        else k-=t[t[u].ch[]].size+,u=t[u].ch[];
    }
}

void write(int u)
{
    pushdown(u);                                //最后输出之前先push_down一遍
    if(t[u].ch[])write(t[u].ch[]);            //先读左儿子
    if(t[u].v>&&t[u].v<n+)printf("%d ",t[u].v-);
    if(t[u].ch[])write(t[u].ch[]);
}

inline void work(int l,int r)            //工作函数
{
    l=get_k(l);
    r=get_k(r+);
    splay(l,);
    splay(r,l);
    t[t[t[root].ch[]].ch[]].mark^=;
}

int main()
{
    n=read();m=read();                    //n个点 m次翻转
    for(int i=;i<=n+;++i)insert(i);    //跳过建树 直接插入即可 再继续看insert()
    for(int i=;i<=m;i++)
    {
        int l=read(),r=read();
        work(l,r);                        //工作函数
    }
    write(root);                        //输出;
    return ;                            //大工告吉
}