BZOJ 1500 维修数列【Splay】

注意：1，内存限制，所以需要回收删除的点

　　　2，当前节点的左连续区间和最大值=max(左子树的左连续区间和最大值，左子树的总和+当节点的值+max(右子树的左连续区间和最大值，0))；右连续区间和最大值同理

　　　　 当前节点的区间和最大值=max(左子树的区间和最大值，max(右子树的区间和最大值，max(左子树的右连续区间和最大值,0)+max(右子树的左连续区间和最大值,0)+当前节点的值))

　　　3，pushdown时，应该交换左子树的左连续区间和最大值与左子树的右连续区间和最大值。因为向上更新时这两个值的不同对结果有影响，不能延迟更新。

#include<stdio.h> 
#include<string.h>
#include<iostream>
using namespace std;
const int N=500100;
const int INF=0x7fffffff;
int data[N],fa[N],id,root; 
int size[N],sum[N],a[N];
int t[N][2],st[N],top;
int mxl[N],mxr[N],mxm[N];
int same[N],flip[N];
void re(int x);
inline void pushup(int x){
    int l=t[x][0],r=t[x][1];
    sum[x]=sum[l]+sum[r]+data[x];
    size[x]=size[l]+size[r]+1;
    mxl[x]=max(mxl[l],sum[l]+data[x]+max(mxl[r],0));
    mxr[x]=max(mxr[r],sum[r]+data[x]+max(mxr[l],0));
    mxm[x]=max(mxm[l],mxm[r]); 
    mxm[x]=max(mxm[x],max(mxr[l],0)+data[x]+max(mxl[r],0));
}
inline void pushdown(int x){
    int l=t[x][0],r=t[x][1];
    if(flip[x]){
        flip[x]=0;
        swap(t[x][0],t[x][1]);
        if(l){
            flip[l]^=1;
            swap(mxl[l],mxr[l]);
        }
        if(r){
            flip[r]^=1;
            swap(mxl[r],mxr[r]);
        }
    }
    if(same[x]!=-INF){
        if(l){
            same[l]=data[l]=same[x];
            sum[l]=same[l]*size[l];
            mxl[l]=mxr[l]=mxm[l]=max(data[l],sum[l]);
        }
        if(r){
            same[r]=data[r]=same[x];
            sum[r]=same[r]*size[r];
            mxl[r]=mxr[r]=mxm[r]=max(data[r],sum[r]);
        }
        same[x]=-INF;
    }
}
void Rotate(int x,int w){
    int y=fa[x],z=fa[y];
    pushdown(y);
    t[y][!w]=t[x][w];
    fa[t[x][w]]=y;
    fa[x]=z;
    t[z][t[z][1]==y]=x;
    t[x][w]=y;
    fa[y]=x;
    pushup(y);
}
void Splay(int x,int y){
    if(x!=y){
        pushdown(x);
        while(fa[x]!=y){
            if(t[fa[x]][0]==x)
                Rotate(x,1);
            else
                Rotate(x,0);
        }
        pushup(x);
        if(!y)root=x;
    }
    
}
void newnode(int &x,int y,int v){
    if(top>0)
        x=st[top--];
    else
        x=++id;
    flip[x]=t[x][0]=t[x][1]=0;
    mxl[x]=mxr[x]=mxm[x]=sum[x]=data[x]=v;
    same[x]=-INF;
    fa[x]=y;
    size[x]=1;
}
void build(int &x,int y,int l,int r){
    if(l<=r){
        int mid=(l+r)>>1;
        newnode(x,y,a[mid]);
        build(t[x][0],x,l,mid-1);
        build(t[x][1],x,mid+1,r);
        pushup(x);
    }
}
void init(int n){
    top=root=id=0;
    t[0][0]=t[0][1]=fa[0]=sum[0]=size[0]=data[0]=flip[0]=0;
    same[0]=mxl[0]=mxr[0]=mxm[0]=-INF;
    newnode(root,0,-INF);
    newnode(t[1][1],1,-INF);
    build(t[2][0],2,1,n);
    pushup(2);
    pushup(1);
}
int Select(int k){
    pushdown(root);
    int x=root;k++;
    for(;size[t[x][0]]+1!=k;){
        if(size[t[x][0]]+1>k)
            x=t[x][0];
        else
            k-=size[t[x][0]]+1,x=t[x][1];
        pushdown(x); 
    }
    return x;
}
void Insert(int c,int n){
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    int cc=Select(c+1);
    c=Select(c);
    Splay(c,0);
    Splay(cc,c);
    build(t[cc][0],cc,1,n);
    pushup(cc);
    pushup(c);
}
void recall(int x){
    if(x){
        st[++top]=x;
        recall(t[x][0]);
        recall(t[x][1]);
    }
}
void Delete(int l,int r){
    r=Select(r+1);
    l=Select(l-1);
    Splay(l,0);
    Splay(r,l);
    recall(t[r][0]);
    t[r][0]=0;
    pushup(r);
    pushup(l);
}
void Flip(int l,int r){
    r=Select(r+1);
    l=Select(l-1);
    Splay(l,0);
    Splay(r,l);
    flip[t[r][0]]^=1;
    swap(mxl[t[r][0]],mxr[t[r][0]]);
}
void Same(int l,int r,int v){
    r=Select(r+1);
    l=Select(l-1);
    Splay(l,0);
    Splay(r,l);
    int x=t[r][0];
    same[x]=data[x]=v;
    sum[x]=data[x]*size[x];
    mxl[x]=mxr[x]=mxm[x]=max(v,sum[x]);
    pushup(r);
    pushup(l);
} 
int Sum(int l,int r){
    r=Select(r+1);
    l=Select(l-1);
    Splay(l,0);
    Splay(r,l);
    return sum[t[r][0]];
}
int Max(int l,int r){
    l=Select(l-1),r=Select(r+1);
    Splay(l,0),Splay(r,l);
    return mxm[t[r][0]];
}
void de(){
    for(int i=1;i<=id;i++){
        printf("%d %d %d %d %d++\n",i,t[i][0],t[i][1],sum[i],mxm[i]);
    }
}
void re(int x){
    if(x){
        re(t[x][0]);
        printf("%d ",data[x]);
        re(t[x][1]);
    }
}
int main(){
    int n,m,i,p,cnt,v;
    char op[20];
    while(scanf("%d%d",&n,&m)!=EOF){
        for(i=1;i<=n;i++)    scanf("%d",&a[i]);
        init(n);
        //de();
        while(m--){
            scanf("%s",op);
            if(!strcmp(op,"GET-SUM")){
                scanf("%d%d",&p,&cnt);
                printf("%d\n",Sum(p,p+cnt-1));
            }
            if(!strcmp(op,"MAX-SUM")){
                printf("%d\n",Max(1,size[root]-2));
            }
            if(!strcmp(op,"INSERT")){
                scanf("%d%d",&p,&cnt);
                Insert(p,cnt);
            }
            if(!strcmp(op,"DELETE")){
                scanf("%d%d",&p,&cnt);
                Delete(p,p+cnt-1);
            }
            if(!strcmp(op,"MAKE-SAME")){
                scanf("%d%d%d",&p,&cnt,&v);
                Same(p,p+cnt-1,v);
            }
            if(!strcmp(op,"REVERSE")){
                scanf("%d%d",&p,&cnt);
                Flip(p,p+cnt-1);
            }
        }
    }
    return 0;
}
/*
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
*/