题解 P3391 【【模板】文艺平衡树(Splay)】

luogu
秀得飞起的咕咕日报
广告


此题函数返回int会MLE???
就像这样:
MLE得飞起
改后是这样:
qwq


#include<bits/stdc++.h>
using namespace std;
int son[1000010][2],val[1000010],cnt[1000010],Fa[1000010],size[1000010],rev[1000010];
int root,ncnt;
int n,m,pd,num;
int Check(int x)
{
    return son[Fa[x]][1]==x;
}
void pushup(int x)
{
    size[x]=size[son[x][0]]+size[son[x][1]]+cnt[x];
}
void pushdown(int x)
{
    if(rev[x])
    {
        swap(son[x][0],son[x][1]);
        rev[son[x][0]]=!rev[son[x][0]];
        rev[son[x][1]]=!rev[son[x][1]];
        rev[x]=0;
    }
}
void rotate(int x)
{
    int y=Fa[x],z=Fa[y],chk=Check(x),tmp=son[x][!chk];
    son[y][chk]=tmp;
    Fa[tmp]=y;
    son[z][Check(y)]=x;
    Fa[x]=z;
    son[x][!chk]=y;
    Fa[y]=x;
    pushup(y);
    pushup(x);
}
void Splay(int x,int goal=0)
{
    for(;Fa[x]!=goal;)
    {
        int y=Fa[x],z=Fa[y];
        if(z!=goal)
        {
            if(Check(x)==Check(y))
            {
                rotate(y);
            }
            else
            {
                rotate(x);
            }
        }
        rotate(x);
    }
    if(!goal)
    {
        root=x;
    }
}
void Find(int x)
{
    if(!root)
    {
        return ;
    }
    int Res=root;
    for(;son[Res][x>val[Res]]&&x!=val[Res];)
    {
        Res=son[Res][x>val[Res]];
    }
    Splay(Res);
}
void insert(int x)
{
    int Res=root,p=0;
    for(;Res&&val[Res]!=x;)
    {
        p=Res;
        Res=son[Res][x>val[Res]];
    }
    if(Res)
    {
    	cnt[Res]++;
    }
    else
    {
    	Res=++ncnt;
        if(p)
        {
        	son[p][x>val[p]]=Res;
        }
        son[Res][0]=0;
        son[Res][1]=0;
        val[Res]=x;
        Fa[Res]=p;
        cnt[Res]=1;
        size[Res]=1;
    }
    Splay(Res);
}
int xth(int x)
{
    int Res=root;
    for(;;)
    {
        pushdown(Res); 
        if(x<=size[son[Res][0]]&&son[Res][0])
        {
            Res=son[Res][0];
        }
        else
        {
            if(x>size[son[Res][0]]+cnt[Res])
            {
                x-=size[son[Res][0]]+cnt[Res];
                Res=son[Res][1];
            }
            else
            {
                return Res;
            }
        }
    }
}
void REV(int L,int R)
{
    int x=xth(L),y=xth(R+2);
    Splay(x);
    Splay(y,x);
    rev[son[y][0]]=!rev[son[y][0]];
}
int RANK(int x)
{
    Find(x);
    return size[son[root][0]];
}
int Pre(int x)
{
    Find(x);
    if(val[root]<x)
    {
        return root;
    }
    int Res=son[root][0]; 
    for(;son[Res][1];)
    {
        Res=son[Res][1];
    }
    return Res;
}
int Suc(int x)
{
    Find(x);
    if(val[root]>x)
    {
        return root;
    }
    int Res=son[root][1]; 
    for(;son[Res][0];)
    {
        Res=son[Res][0];
    }
    return Res;
}
void Delete(int x)
{
    int las=Pre(x),nex=Suc(x);
    Splay(las);
    Splay(nex,las);
    int DEL=son[nex][0];
    if(cnt[DEL]>1)
    {
        cnt[DEL]--;
        Splay(DEL);
    }
    else
    {
        son[nex][0]=0;
    } 
}
void output(int x)
{
    pushdown(x);
    if (son[x][0])
        output(son[x][0]);
    if(val[x]&&val[x]<=n) 
        printf("%d ", val[x]);
    if(son[x][1])
        output(son[x][1]);
}
int main()
{
    cin>>n>>m;
    for(int i=0;i<=n+1;i++)
    {
        insert(i);
    }
    int ipl,ipr; 
    for(int i=1;i<=m;i++)
    {
        cin>>ipl>>ipr;
        REV(ipl,ipr);
    }
    output(root);
}
posted @ 2019-03-22 21:36  G_A_TS  阅读(381)  评论(0编辑  收藏  举报