bzoj 1500: [NOI2005]维修数列 -- splay

1500: [NOI2005]维修数列

Time Limit: 10 Sec  Memory Limit: 64 MB

Description

Input

输入的第1 行包含两个数N 和M(M ≤20 000)，N 表示初始时数列中数的个数，M表示要进行的操作数目。
第2行包含N个数字，描述初始时的数列。
以下M行，每行一条命令，格式参见问题描述中的表格。
任何时刻数列中最多含有500 000个数，数列中任何一个数字均在[-1 000, 1 000]内。
插入的数字总数不超过4 000 000个，输入文件大小不超过20MBytes。

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

 

 

Source

splay 练手题？

第一次打比较全的splay

照着黄学长的代码打了一份qaq。。。

#include<map>
#include<cmath>
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define inf 1007
#define ll long long
#define N 500010
inline int rd()
{
    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;
}
int dd[N],rt,c[N][2],sum[N],mx[N],v[N],sz[N];
bool rev[N],tag[N];
int fa[N],lx[N],rx[N];
int n,m,a[N],tot;
char s[10];
queue<int>q;
#define ls c[x][0]
#define rs c[x][1]
inline void upd(int x)
{
    if(!x)return;
    sum[x]=sum[ls]+sum[rs]+v[x];
    sz[x]=sz[ls]+sz[rs]+1;
    mx[x]=max(mx[ls],mx[rs]);
    mx[x]=max(mx[x],rx[ls]+v[x]+lx[rs]);
    lx[x]=max(lx[ls],sum[ls]+v[x]+lx[rs]);
    rx[x]=max(rx[rs],sum[rs]+v[x]+rx[ls]);
}
inline void pd(int x)
{
    if(tag[x])
    {
        tag[x]=rev[x]=0;
        if(ls) tag[ls]=1,v[ls]=v[x],sum[ls]=v[x]*sz[ls];
        if(rs) tag[rs]=1,v[rs]=v[x],sum[rs]=v[x]*sz[rs];
        if(v[x]>=0)
        {
            if(ls)lx[ls]=rx[ls]=mx[ls]=sum[ls];
            if(rs)lx[rs]=rx[rs]=mx[rs]=sum[rs];
        }
        else 
        {
            if(ls) lx[ls]=rx[ls]=0,mx[ls]=v[x];
            if(rs) lx[rs]=rx[rs]=0,mx[rs]=v[x];
        }
    }
    if(rev[x])
    {
        rev[x]=0;rev[ls]^=1;rev[rs]^=1;
        swap(lx[ls],rx[ls]);swap(lx[rs],rx[rs]);
        swap(c[ls][0],c[ls][1]);
        swap(c[rs][0],c[rs][1]);
    }
}
int build(int l,int r)
{
    if(l>r) return 0;
    int mid=l+r>>1,x=dd[mid];
    if(l==r)
    {
        mx[x]=sum[x]=a[l];sz[x]=1;
        lx[x]=rx[x]=(a[l]>0?a[l]:0);
    }
    else{
        ls=build(l,mid-1);
        rs=build(mid+1,r);
        fa[ls]=fa[rs]=x;
    }
    v[x]=a[mid];upd(x);
    return x;
}
int fd(int x,int k)
{
    pd(x);
    if(sz[ls]>=k) return fd(ls,k);
    if(sz[ls]+1==k) return x;
    return fd(rs,k-sz[ls]-1);
}
void rot(int x,int &k)
{
    int y=fa[x],z=fa[y],l,r;
    l=(c[y][1]==x);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;fa[c[x][r]]=y;
    c[y][l]=c[x][r];c[x][r]=y;
    upd(y);upd(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)) rot(x,k);
            else rot(y,k);
        }
        rot(x,k);
    }
}
int split(int k,int t)
{
    int x=fd(rt,k),y=fd(rt,k+t+1);
    splay(x,rt);splay(y,rs);
    return c[y][0];
}
void rec(int x)
{
    if(!x) return;
    rec(ls);rec(rs);
    q.push(x);
    fa[x]=ls=rs=0;
    rev[x]=tag[x]=0;
}
inline void xg(int k,int t,int V)
{
    int x=split(k,t),y=fa[x];
    v[x]=V;tag[x]=1;sum[x]=sz[x]*V;
    if(V>=0) lx[x]=rx[x]=mx[x]=sum[x];
    else lx[x]=rx[x]=0,mx[x]=V;
    upd(y);upd(fa[y]);
}
void rever(int k,int t)
{
    int x=split(k,t),y=fa[x];
    if(!tag[x])
    {
        rev[x]^=1;
        swap(ls,rs);
        swap(lx[x],rx[x]);
        upd(y);upd(fa[y]);
    }
}
int main()
{
    n=rd();m=rd();
    for(int i=1;i<=n;i++) a[i+1]=rd();
    for(int i=1;i<=n+2;i++) dd[i]=i;
    mx[0]=a[1]=a[n+2]=-inf;
    rt=build(1,n+2);tot=n+2;
    int x,t,y,z,k;
    while(m--)    
    {
        scanf("%s",s);
        if(s[0]=='I') //INSERT
        {
            k=rd();t=rd();
            for(int i=1;i<=t;i++) a[i]=rd();
            for(int i=1;i<=t;i++)
            {
                if(!q.empty()) dd[i]=q.front(),q.pop();
                else dd[i]=++tot;
            }
            z=build(1,t);
            x=fd(rt,k+1);y=fd(rt,k+2);
            splay(x,rt);splay(y,rs);
            fa[z]=y;c[y][0]=z;
            upd(y);upd(x);
        }
        else if(s[0]=='D') //DELETE
        {
            k=rd();t=rd();
            x=split(k,t);y=fa[x];
            rec(x);c[y][0]=0;
            upd(y);upd(fa[y]);
        }
        else if(s[0]=='M')
        {
            if(s[2]=='X')//MAX-SUM
            {    
                printf("%d\n",mx[rt]);
            }
            else //MAKE-SAME
            {
                k=rd();t=rd();x=rd();
                xg(k,t,x);
            }
        }
        else if(s[0]=='G')//GET-SUM
        {
            k=rd();t=rd();
            x=split(k,t);
            printf("%d\n",sum[x]);
        }
        else //REVERSE
        {
            k=rd();t=rd();
            rever(k,t);
        }
    }
    return 0;
}