【NOI2005】维护数列

最新推荐文章于 2026-04-18 03:01:30 发布

原创最新推荐文章于 2026-04-18 03:01:30 发布 · 1.6k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

数据结构同时被 2 个专栏收录

50 篇文章

订阅专栏

splay

3 篇文章

订阅专栏

本文介绍了如何使用Splay树解决一个维护数列的问题，包括插入、删除、修改、翻转、求和及求和最大子列等6种操作。分析中提到，Splay树可以有效处理区间和、最大子列和等信息，通过标记处理翻转和修改操作，利用栈管理空间以适应数列数量的变化。

Description

请写一个程序，要求维护一个数列，支持以下6种操作：（请注意，格式栏中的下划线‘ _ ’表示实际输入文件中的空格）
1. 插入 INSERT_posi_tot_c1_c2_…_ctot 在当前数列的第posi个数字后插入tot个数字：c1, c2, …, ctot；若在数列首插入，则posi为0
2. 删除 DELETE_posi_tot 从当前数列的第posi个数字开始连续删除tot个数字
3. 修改 MAKE-SAME_posi_tot_c 将当前数列的第posi个数字开始的连续tot个数字统一修改为c
4. 翻转 REVERSE_posi_tot 取出从当前数列的第posi个数字开始的tot个数字，翻转后放入原来的位置
5. 求和 GET-SUM_posi_tot 计算从当前数列开始的第posi个数字开始的tot个数字的和并输出
6. 求和最大的子列 MAX-SUM 求出当前数列中和最大的一段子列，并输出最大和

Input

输入文件的第1行包含两个数N和M，N表示初始时数列中数的个数，M表示要进行的操作数目。
第2行包含N个数字，描述初始时的数列。
以下M行，每行一条命令，格式参见问题描述中的表格。

Output

对于输入数据中的GET-SUM和MAX-SUM操作，向输出文件依次打印结果，每个答案（数字）占一行。

Sample Input

9 8
2 -6 3 5 1 -5 -3 6 3
GET-SUM 5 4
MAX-SUM
INSERT 8 3 -5 7 2
DELETE 12 1
MAKE-SAME 3 3 2
REVERSE 3 6
GET-SUM 5 4
MAX-SUM

Sample Output

-1
10
1
10

Data Constraint

你可以认为在任何时刻，数列中至少有1个数。
输入数据一定是正确的，即指定位置的数在数列中一定存在。
50%的数据中，任何时刻数列中最多含有30 000个数；
100%的数据中，任何时刻数列中最多含有500 000个数。
100%的数据中，任何时刻数列中任何一个数字均在[-1 000, 1 000]内。
100%的数据中，M ≤20 000，插入的数字总数不超过4 000 000个，输入文件大小不超过20MBytes

分析

这种带翻转、插入、删除操作的维护序列的问题，很容易想到splay

用splay维护区间[l,r]的以下几个数：
sum（表示区间和）
ls（表示以l为左端的子列的和的最大值）
rs（表示以r为右端的子列的和的最大值）
ms（表示此区间所有子列的和的最大值）

然后看每个操作：

插入、删除和一般没什么区别，先选取区间，然后操作。

make_same：选取区间，然后打个标记。

翻转：同样是打标记。下传时，把对应区间的ls和rs交换，其它不变（因为翻转后，维护的ls,rs,ms,sum对应的是该区间从右到左遍历出来的数列，其中总和、ms是不会变的，ls,rs也只是换到了另一边）

求和：也是和一般的没什么区别。

求和最大的子列：当在splay旋转时进行update(x)操作时，我们把节点son[x][0],x,son[x][1] 分别看成一个区间，先合并son[x][0]和x，再与son[x][1]合并。注意：要把区间[1..n]（n为当前序列数的个数）选取出来再输出（或者把维护序列时前后加的数赋值为一个很小的数），否则会影响答案。

空间

开4000000的数组会爆空间，但是题目限制的每一时刻数的数量最多为500000，那么可以回收删除操作后多出的空间。具体是用一个栈来存当前可用的空间，插入时从栈顶弹出，删除时加入栈。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn=500010,Null=500002;

int m,n,root,top,son[maxn][2],fa[maxn],mask[maxn],sum[maxn],size[maxn],ls[maxn],rs[maxn],ms[maxn],a[maxn],st[maxn],A[maxn],data[maxn];

bool flip[maxn],bz;

char c;

int read()
{
    int x=bz=0;
    for (c=getchar();(c<'0' || c>'9') && c!='-';c=getchar());
    if (c=='-')
    {
        bz=1; c=getchar();
    }
    for (;c>='0' && c<='9';c=getchar()) x=x*10+c-48;
    if (bz) x=-x;
    return x;
}

void make_same(int x,int y)
{
    a[x]=mask[x]=y; sum[x]=y*size[x];
    if (y>0) ls[x]=rs[x]=ms[x]=sum[x];//子列不能为空
    else ls[x]=rs[x]=ms[x]=y;
}

void push_down(int x)//下传标记
{
    if (flip[x])
    {
        son[x][0]^=son[x][1]^=son[x][0]^=son[x][1];
        flip[son[x][0]]^=1; flip[son[x][1]]^=1;
        ls[son[x][0]]^=rs[son[x][0]]^=ls[son[x][0]]^=rs[son[x][0]];
        ls[son[x][1]]^=rs[son[x][1]]^=ls[son[x][1]]^=rs[son[x][1]];
        flip[x]=0;
    }
    if (mask[x]!=Null)
    {
        if (son[x][0]!=Null) make_same(son[x][0],mask[x]);
        if (son[x][1]!=Null) make_same(son[x][1],mask[x]);
        mask[x]=Null;
    }
}

void calc(int ll,int lr,int lm,int ls,int rl,int rr,int rm,int rs,int &l,int &r,int &m)
{
    l=max(ll,ls+rl);
    r=max(rr,rs+lr);
    m=max(max(lm,rm),lr+rl);
}//合并两个区间的ls,rs,ms

void update(int x)
{
    if (son[x][0]==Null)
    {
        if (son[x][1]==Null)
        {
            ls[x]=rs[x]=ms[x]=a[x];
        }else calc(a[x],a[x],a[x],a[x],ls[son[x][1]],rs[son[x][1]],ms[son[x][1]],sum[son[x][1]],ls[x],rs[x],ms[x]);
    }else
    {
        if (son[x][1]==Null)
        calc(ls[son[x][0]],rs[son[x][0]],ms[son[x][0]],sum[son[x][0]],a[x],a[x],a[x],a[x],ls[x],rs[x],ms[x]);
        else
        {
            calc(ls[son[x][0]],rs[son[x][0]],ms[son[x][0]],sum[son[x][0]],a[x],a[x],a[x],a[x],ls[x],rs[x],ms[x]);
            calc(ls[x],rs[x],ms[x],sum[son[x][0]]+a[x],ls[son[x][1]],rs[son[x][1]],ms[son[x][1]],sum[son[x][1]],ls[x],rs[x],ms[x]);
        }
    }
    sum[x]=sum[son[x][0]]+sum[son[x][1]]+a[x];
    size[x]=size[son[x][0]]+size[son[x][1]]+1;
}

void Rotate(int x,int t)
{
    int y=fa[x];
    if (y!=root)
    {
        if (son[fa[y]][0]==y) son[fa[y]][0]=x;else son[fa[y]][1]=x;
    }else root=x;
    fa[x]=fa[y]; fa[y]=x; son[y][t]=son[x][1-t]; son[x][1-t]=y; if (son[y][t]!=Null) fa[son[y][t]]=y;
    update(y);
}

void splay(int x,int f)
{
    while (fa[x]!=f)
    {
        int y=fa[x];
        if (fa[y]==f)
        {
            push_down(y); push_down(x);
            if (son[y][0]==x) Rotate(x,0); else Rotate(x,1);
        }else
        {
            int z=fa[y];
            push_down(z); push_down(y); push_down(x);
            if (son[y][0]==x)
            {
                if (son[z][0]==y)
                {
                    Rotate(y,0); Rotate(x,0);
                }else
                {
                    Rotate(x,0); Rotate(x,1);
                }
            }else
            {
                if (son[z][0]==y)
                {
                    Rotate(x,1); Rotate(x,0);
                }else
                {
                    Rotate(y,1); Rotate(x,1);
                }
            }
        }
    }
    update(x);
}

void Select(int x,int f)
{
    int i=root;
    while (size[son[i][0]]!=x)
    {
        if (size[son[i][0]]>x) i=son[i][0];
        else
        {
            x-=size[son[i][0]]+1; i=son[i][1];
        }
        push_down(i);
    }
    splay(i,f);
}

void Delete(int x,int t)
{
    int tot=1; data[1]=son[x][t];
    for (int i=1;i<=tot;i++)
    {
        st[++top]=data[i];//回收
        if (son[data[i]][0]!=Null) data[++tot]=son[data[i]][0];
        if (son[data[i]][1]!=Null) data[++tot]=son[data[i]][1];
    }
    son[x][t]=Null;
    splay(x,Null);
}

void build(int l,int r,int &root)
{
    int mid=(l+r)/2;
    a[root=st[top--]]=A[mid];//在栈顶得到它的编号
    if (l<mid)
    {
        build(l,mid-1,son[root][0]);
        fa[son[root][0]]=root;
    }else son[root][0]=Null;
    if (mid<r)
    {
        build(mid+1,r,son[root][1]);
        fa[son[root][1]]=root;
    }else son[root][1]=Null;
    update(root); mask[root]=Null; flip[root]=0;
}

int main()
{
    n=read(); m=read();
    for (int i=Null-1;i>=0;i--) st[++top]=i;
    for (int i=1;i<=n;i++) A[i]=read();
    build(0,n+1,root); fa[root]=Null;
    while (m--)
    {
        for (c=getchar();c<'A' || c>'R';c=getchar());
        if (c=='G')
        {
            for (c=getchar();c!='-';c=getchar());
            int l=read(),tot=read(),r=l+tot;
            Select(l-1,Null); Select(r,root);
            printf("%d\n",sum[son[son[root][1]][0]]);
        }else if (c=='I')
        {
            int x=read(),tot=read(); n+=tot;
            for (int i=1;i<=tot;i++) A[i]=read();
            Select(x,Null); Select(x+1,root);
            build(1,tot,son[son[root][1]][0]); fa[son[son[root][1]][0]]=son[root][1];
            splay(son[son[root][1]][0],Null);
        }else if (c=='D')
        {
            int l=read(),tot=read(),r=l+tot; n-=tot;
            Select(l-1,Null); Select(r,root);
            Delete(son[root][1],0);
        }else if (c=='R')
        {
            int l=read(),tot=read(),r=l+tot;
            Select(l-1,Null); Select(r,root);
            int t=son[son[root][1]][0];
            flip[t]^=1;
            ls[t]^=rs[t]^=ls[t]^=rs[t];
            splay(t,Null);
        }else
        {
            c=getchar(); c=getchar();
            if (c=='K')
            {
                for (c=getchar();c!='-';c=getchar());
                int l=read(),tot=read(),r=l+tot,C=read();
                Select(l-1,Null); Select(r,root);
                make_same(son[son[root][1]][0],C);
                splay(son[son[root][1]][0],Null);
            }else
            {
                for (c=getchar();c!='-';c=getchar());
                for (c=getchar();c>='A' && c<='Z';c=getchar());
                Select(0,Null); Select(n+1,root);//
                printf("%d\n",ms[son[son[root][1]][0]]);
            }
        }
    }
    return 0;
}