【bzoj1500】[NOI2005]维修数列

Description

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

HINT

题解
splay基本操作，参见黄学长博客（本人代码基本仿照）

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<cmath>
#include<algorithm>
#define inf 1000000000
#define N 1000005
using namespace std;
int n,m,rt,cnt;
int a[N],id[N],fa[N],c[N][2];
int sum[N],size[N],v[N],mx[N],lx[N],rx[N];
bool tag[N],rev[N];
queue<int> q;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void update(int x)
{
    int l=c[x][0],r=c[x][1];
    sum[x]=sum[l]+sum[r]+v[x];
    size[x]=size[l]+size[r]+1;
    mx[x]=max(mx[l],mx[r]);
    mx[x]=max(mx[x],rx[l]+v[x]+lx[r]);
    lx[x]=max(lx[l],sum[l]+v[x]+lx[r]);
    rx[x]=max(rx[r],sum[r]+v[x]+rx[l]);
}

void pushdown(int x)
{
    int l=c[x][0],r=c[x][1];
    if(tag[x])
    {
        rev[x]=tag[x]=0;
        if(l)tag[l]=1,v[l]=v[x],sum[l]=v[x]*size[l];
        if(r)tag[r]=1,v[r]=v[x],sum[r]=v[x]*size[r];
        if(v[x]>=0)
        {
            if(l)lx[l]=rx[l]=mx[l]=sum[l];
            if(r)lx[r]=rx[r]=mx[r]=sum[r];
        }
        else 
        {
            if(l)lx[l]=rx[l]=0,mx[l]=v[x];
            if(r)lx[r]=rx[r]=0,mx[r]=v[x];
        }
    }
    if(rev[x])
    {
        rev[x]^=1;rev[l]^=1;rev[r]^=1;
        swap(lx[l],rx[l]);swap(c[l][0],c[l][1]);
        swap(lx[r],rx[r]);swap(c[r][0],c[r][1]);
    }
}

void rotate(int x,int &k)
{
    int l,r,y=fa[x],z=fa[y];
    if (c[y][0]==x)l=0;else l=1;r=l^1;
    if (y==k)k=x;
    else if (c[z][0]==y)c[z][0]=x;else c[z][1]=x;
    fa[x]=z;fa[y]=x;
    c[y][l]=c[x][r];
    fa[c[x][r]]=y;c[x][r]=y;
    update(y);update(x);
}

void splay(int x,int &k)
{
    int y,z;
    while (x!=k)
    {
        y=fa[x];z=fa[y];
        if (y!=k)
        {
            if (c[y][0]==x^c[z][0]==y)rotate(x,k);
            else rotate(y,k);
        }
        rotate(x,k);
    }
}

int find(int k,int x)
{
    pushdown(k);
    if (size[c[k][0]]+1==x) return k;
    else if (size[c[k][0]]>=x) return find(c[k][0],x);
    else return find(c[k][1],x-size[c[k][0]]-1);
}

void build(int l,int r,int f)
{
    if (l>r) return;
    int mid=(l+r)/2,now=id[mid],last=id[f];
    if (l==r)
    {
        sum[now]=a[l];size[now]=1;tag[now],rev[now]=0;
        if(a[l]>=0)lx[now]=rx[now];
        else lx[now]=rx[now]=0;
        mx[now]=a[l];
    }
    else build(l,mid-1,mid),build(mid+1,r,mid);
    fa[now]=last;v[now]=a[mid];
    update(now);
    c[last][mid>=f]=now;
}

void insert(int k,int tot)
{
    for (int i=1;i<=tot;i++)a[i]=read();
    for (int i=1;i<=tot;i++)
    {
        if (!q.empty())
        {
            id[i]=q.front();q.pop();
        }
        else id[i]=++cnt;
    }
    build(1,tot,0);int z=id[(1+tot)>>1];
    int x=find(rt,k+1),y=find(rt,k+2);
    splay(x,rt);splay(y,c[x][1]);
    fa[z]=y;c[y][0]=z;
    update(y);update(x);
}

void rec(int x)
{
    if(!x)return;
    int l=c[x][0],r=c[x][1];
    rec(l);rec(r);q.push(x);
    fa[x]=c[x][0]=c[x][1]=0;
    tag[x]=rev[x]=0;
}

void del(int k,int tot)
{
    int x=find(rt,k),y=find(rt,k+tot+1);
    splay(x,rt);splay(y,c[x][1]);
    rec(c[y][0]);c[y][0]=0;
    update(y);update(x);
}

void rever(int k,int tot)
{
    int x=find(rt,k),y=find(rt,k+tot+1);
    splay(x,rt);splay(y,c[x][1]);
    if (!tag[c[y][0]])
    {
        int k=c[y][0];rev[k]^=1;
        swap(lx[k],rx[k]);swap(c[k][0],c[k][1]);    
        update(y);update(x);
    }
}

void query_sum(int k,int tot)
{
    int x=find(rt,k),y=find(rt,k+tot+1);
    splay(x,rt);splay(y,c[x][1]);
    printf("%d\n",sum[c[y][0]]);
}

void change(int st,int tot,int val)
{
    int x=find(rt,st),y=find(rt,st+tot+1);
    splay(x,rt);splay(y,c[x][1]);
    int k=c[y][0];
    v[k]=val;
    tag[k]=1;
    sum[k]=size[k]*val;
    if(val>=0)lx[k]=rx[k]=mx[k]=sum[k];
    else lx[k]=rx[k]=0,mx[k]=val;
    update(y);update(x);
}

int main()
{
    n=read();m=read();
    mx[0]=a[1]=a[n+2]=-inf;
    for (int i=2;i<=n+1;i++)a[i]=read();
    for (int i=1;i<=n+2;i++)id[i]=i;
    build(1,n+2,0);cnt=n+2;rt=(n+3)>>1;
    char s[10];int k,tot,val;
    while (m--)
    {   
        scanf("%s",s);
        if(s[0]!='M'||s[2]!='X')k=read(),tot=read();
        if (s[0]=='I')insert(k,tot);
        if (s[0]=='D')del(k,tot);
        if (s[0]=='R')rever(k,tot);
        if (s[0]=='G')query_sum(k,tot);
        if (s[0]=='M')
        {
            if (s[2]=='X')printf("%d\n",mx[rt]);
            else
            {
                val=read();change(k,tot,val);
            }
        }
    }
    return 0;
}