维护数列

题目大意

维护一段数列,支持添加,删除,翻转,区间赋值,询问最大子段和以及区间和等操作。

splay

我的代码常数较大,被卡了。。。

#include
#include
#include
#include
#define fo(i,a,b) for(i=a;i<=b;i++)
#define ll long long
using namespace std;
const int maxn=6e5+5;
int n,m,i,root,b,w[maxn],bz[maxn],q[maxn],fa[maxn],s[maxn],tree[maxn][2];
bool bf[maxn];
struct ar{
    int a,l,r,ma,sum;
    ar(int x=0){l=r=ma=a=sum=x;}
}d[maxn];
int read(){
    int n=0,f=1;char c=getchar();
    while (((c<'0')||(c>'9'))&&(c!='-')) c=getchar();
    if (c=='-') {
        c=getchar();
        if ((c>='0')&&(c<='9')) f=-1;else return read();
    }
    while ((c>='0')&&(c<='9')) n=n*10+c-'0',c=getchar();
    return n*f;
}
int pd(int x){return (tree[fa[x]][0]==x)?0:1;}
void update(int x){
    int l=tree[x][0],r=tree[x][1],l1=max(0,d[r].l),r1=max(0,d[l].r);
    d[x].sum=d[x].a+d[l].sum+d[r].sum;
    d[x].l=max(d[l].l,d[l].sum+d[x].a+l1);
    d[x].r=max(d[r].r,d[r].sum+d[x].a+r1);
    d[x].ma=max(d[l].ma,max(d[r].ma,r1+d[x].a+l1));
    s[x]=s[tree[x][0]]+s[tree[x][1]]+1;
}
void rotate(int x){
    int y=fa[x],z=pd(x),z1=pd(y);
    if (tree[y][z]=tree[x][1-z]) fa[tree[y][z]]=y;
    if (fa[x]=fa[y]) tree[fa[x]][z1]=x;
    fa[y]=x,tree[x][1-z]=y;
    update(y);
}
void ch(int x,int y){
    if (!x) return;
    d[x].a=bz[x]=y;
    d[x].l=d[x].r=d[x].ma=d[x].sum=s[x]*y;
    if (y<0) d[x].l=d[x].r=d[x].ma=y;
}
void re(int x){
    if (!x) return;
    swap(tree[x][0],tree[x][1]);
    swap(d[x].l,d[x].r);
    bf[x]^=1;
}
void clear(int x){
    int l=tree[x][0],r=tree[x][1];
    if (bz[x]!=b) ch(l,bz[x]),ch(r,bz[x]);
    if (bf[x]) re(l),re(r);
    bf[x]=0,bz[x]=b;
}
void chu(int x,int y){
    q[0]=0;
    while (x!=y){
        q[++q[0]]=x;
        x=fa[x];
    }
    while (q[0]) clear(q[q[0]]),q[0]--;
}
void splay(int x,int y){
    chu(x,y);
    while (fa[x]!=y){
        int f=fa[x];
        if (fa[f]!=y) 
        if (pd(f)==pd(x)) rotate(f);
        rotate(x);
    }
    update(x);
}
int kth(int x,int y){
    clear(x);
    int sl=s[tree[x][0]];
    if (sl+1==y) return x;
    if (sl>=y) return kth(tree[x][0],y);
    return kth(tree[x][1],y-sl-1);
}
void merge(int x,int y,int &l){
    if ((!x)||(!y)) {l=x+y;return;}
    l=kth(x,s[x]);
    splay(l,0);
    tree[l][1]=y,fa[y]=l;
    update(l);
}
void del(int x){
    x=kth(root,x),splay(x,0);
    w[++w[0]]=x;
    fa[tree[x][0]]=fa[tree[x][1]]=0;
    merge(tree[x][0],tree[x][1],root);
    tree[x][0]=tree[x][1]=0;
}
void split(int x,int p,int &l,int &r){
    if (!p){
        l=0,r=x;
        return;
    }
    l=kth(x,p);
    splay(l,0);
    fa[r=tree[l][1]]=0,tree[l][1]=0;
    update(l);
}
void ins(int x,int y){
    int p;
    if (w[0]) p=w[w[0]],w[0]--;else p=++n;
    d[p]=ar(y),s[p]=1;
    int l,r;
    split(root,x,l,r);
    merge(l,p,l),merge(l,r,root);
}
char get(){
    char c=getchar();
    while ((c<'A')||(c>'Z')) c=getchar();
    return c;
}
void bu(int a,int b,int &x){
    if (a==b) {
        x=a,s[a]=1;
        return;
    }int m=(a+b)/2,l,r;
    bu(a,m,l),bu(m+1,b,r);
    merge(l,r,x);
}
int main(){
    memset(bz,127/3,sizeof(bz));b=bz[0];
    n=read(),m=read();
    d[0].l=d[0].r=d[0].ma=-b;
    fo(i,1,n) d[i]=ar(read());
    bu(1,n,root);
    while(m--){
        char c=get();
        if (c=='I'){
            int x=read(),t=x+read(),y;
            while (xy=read(),ins(x,y),x++;
            continue;
        }
        if (c=='D') {
            int x=read(),t=read();
            while (t--)del(x);
            continue;
        }
        int x,y,z,l,r,l1;
        if (c=='M') {
            c=get(),c=get();
            if (c=='K') {
                x=read(),y=read(),z=read();
                split(root,x-1,l,r),split(r,y,l1,r);
                ch(l1,z);
                merge(l1,r,r),merge(l,r,root);
            }else {
                while (((c>='A')&&(c<='Z'))||(c=='-')) c=getchar();
                printf("%d\n",d[root].ma);
            }continue;
        }
        x=read(),y=read();
        split(root,x-1,l,r),split(r,y,l1,r);
        if (c=='R') re(l1);
        if (c=='G') printf("%d\n",d[l1].sum);
        merge(l1,r,r),merge(l,r,root);
    }
}

你可能感兴趣的:(splay)