(模板)Splay维护区间序列(洛谷P3391)

题目:

您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作:翻转一个区间,例如原有序序列是5 4 3 2 1,翻转区间是[2,4]的话,结果是5 2 3 4 1
输入格式:
第一行为n,m n表示初始序列有n个数,这个序列依次是1,2,3,……,n, m表示翻转操作次数
接下来m行每行两个数 [l,r][l,r] 数据保证 1≤l≤r≤n
输出格式:
输出一行n个数字,表示原始序列经过m次变换后的结果

详细解释:https://blog.csdn.net/A_Comme_Amour/article/details/79382104

#include
#include
#include
#include
using namespace std;
const int MAXN=1000007;
const int inf=1e9;

int f[MAXN],cnt[MAXN],ch[MAXN][2],size[MAXN],key[MAXN],tag[MAXN],sz,rt;
int n,m,x,y,data[MAXN];

bool get(int x)
{
    return ch[f[x]][1]==x;
}

void update(int x)
{
    size[x]=size[ch[x][0]]+size[ch[x][1]]+1;
}

void pushdown(int x)//下传反转操作
{
    if (x && tag[x])
    {
        tag[ch[x][0]]^=1;
        tag[ch[x][1]]^=1;
        swap(ch[x][0],ch[x][1]);
        tag[x]=0;
    }
}

void rotate(int x)//与普通splay的唯一不同就是第二行加了两个pushdown
{
    int old=f[x],oldf=f[old],which=get(x);
    pushdown(old); pushdown(x);//不要忘记pushdown
    ch[old][which]=ch[x][which^1]; f[ch[old][which]]=old;
    ch[x][which^1]=old; f[old]=x;
    f[x]=oldf;
    if (oldf) ch[oldf][ch[oldf][1]==old]=x;
    update(old);
    update(x);
}

void splay(int x,int goal)//比普通的goal加了一个目标goal而已
{
    for (int fa; (fa=f[x])!=goal; rotate(x))
        if (f[fa]!=goal)
            rotate((get(x)==get(fa))?fa:x);
    if (!goal) rt=x;
}

int build_tree(int fa,int l,int r)//类似线段树建树
{
    if (l>r) return 0;
    int mid=(l+r)>>1;
    int now=++sz;
    key[now]=data[mid]; f[now]=fa; tag[now]=0;
    ch[now][0]=build_tree(now,l,mid-1);
    ch[now][1]=build_tree(now,mid+1,r);
    update(now);
    return now;
}

int find(int x)
{
    int now=rt;
    while (1)
    {
        pushdown(now);
        if (x<=size[ch[now][0]]) now=ch[now][0];
        else {
            x-=size[ch[now][0]]+1;
            if (!x) return now;
            now=ch[now][1];
        }
    }
}

void turn(int l,int r)//反转操作
{
    l= find(l);
    r= find(r + 2);
    splay(l,0);//先让l占下根的位置,然后让r+2把他挤到左子树上去
    splay(r,l);
    pushdown(rt);
    tag[ch[ch[rt][1]][0]]^=1;//根的右子树的左子树
}

void write(int now)//中序遍历
{
    pushdown(now);
    if (ch[now][0]) write(ch[now][0]);
    if (key[now]!=-inf && key[now]!=inf) printf("%d ",key[now]);
    if (key[ch[now][1]]) write(ch[now][1]);
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1; i<=n; i++) data[i+1]=i;
    data[1]=-inf; data[n+2]=inf;
    rt=build_tree(0,1,n+2);
    for (int i=1; i<=m; i++) {
        scanf("%d%d", &x, &y);
        turn(x, y);
    }
    write(rt);

    return 0;
}

你可能感兴趣的:(平衡树,Splay,模板,数据结构)