1500: [NOI2005]维修数列(Splay)

Description

请写一个程序,要求维护一个数列,支持以下 6 种操作:
请注意,格式栏 中的下划线‘ _ ’表示实际输入文件中的空格
在这里插入图片描述
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操作,向输出文件依次打印结果,每个答案(数字)占一行。


题解:

建议做这道题之前先去把Can you answer these queries III文艺平衡树都做一下

双倍经验:Can you answer these queries III

1.build 建树:将原序列用类似线段树的写法,整体建树,即不断的将当前点维护的区间进行二分,到达单个元素的区间后,就插入序列值。这样,最初建的树就是非常平衡的。

2.Insert 插入:考虑类似于build的插入,首先将要插入的区间变成一个如上的平衡树,然后把k+1移动到根,k+2移动到跟到右儿子上(因为我们插入了两个哨兵,所以要+1),然后把要Insert插入的树,直接挂到k+2的左儿子即可

3.delete 删除:删除k后面的tot个数,我们可以提取区间(k,k+tot+1),只需要把k结点伸展到根结点,k+tot+1伸展到根节点的右儿子,然后把k+tot+1的左儿子全部清0即可

4.make_same 修改:操作类似,将区间split提取出来,给根结点的右儿子的左儿子打上修改标记即可

5.reverse 修改:和文艺平衡树一样,注意修改标记的优先级>翻转标记

6 get_sum 求和:一样将区间提取出来,然后直接输出根结点的右儿子的左儿子记录的和即可

7 max_sum 求和最大子序列:每个结点维护左连续最大和lx,右连续最大和rx,最大和mx,区间总和sum,然后合并时更新:

mx[x]=max(lx[r]+rx[l]+val[x],max(mx[l],mx[r]));
//最大和mx考虑左右区间最大和,或者这个最大和经过了中间点
//即左区间右连续最大和+右区间左连续最大和+自己的值
lx[x]=max(lx[l],sum[l]+lx[r]+val[x]);
//合并后的左区间最大和:原本左区间的左连续最大和,或者原本左区间总和+右连续左区间和
rx[x]=max(rx[r],sum[r]+rx[l]+val[x]);//同理

AC代码:

#pragma GCC optimize(2)
#include<bits/stdc++.h>
#include<ext/rope>
using namespace std;
using namespace __gnu_cxx;
typedef long long LL;
const int MAXN = 5e5+10;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
int n,m,root,cnt;
int a[MAXN],id[MAXN],fa[MAXN],son[MAXN][2];
int sum[MAXN],sz[MAXN],val[MAXN],mx[MAXN],lx[MAXN],rx[MAXN];
bool tag[MAXN],rev[MAXN]; queue<int> que;
inline int is(int x){ return x==son[fa[x]][1]; }
inline void pushup(int x){
    int l=son[x][0],r=son[x][1];
    sum[x]=sum[l]+sum[r]+val[x];
    sz[x]=sz[l]+sz[r]+1;
    mx[x]=max(lx[r]+rx[l]+val[x],max(mx[l],mx[r]));
    lx[x]=max(lx[l],sum[l]+lx[r]+val[x]);
    rx[x]=max(rx[r],sum[r]+rx[l]+val[x]);
}
inline void pushdown(int x){
    int l=son[x][0],r=son[x][1];
    if(tag[x]){
        rev[x]=tag[x]=0;
        if(l) tag[l]=1,val[l]=val[x],sum[l]=val[x]*sz[l];
        if(r) tag[r]=1,val[r]=val[x],sum[r]=val[x]*sz[r];
        if(val[x]>=0){
            if(l) mx[l]=rx[l]=lx[l]=sum[l];
            if(r) mx[r]=rx[r]=lx[r]=sum[r];
        }else{
            if(l) lx[l]=rx[l]=0,mx[l]=val[x];
            if(r) lx[r]=rx[r]=0,mx[r]=val[x];
        }
    }
    if(rev[x]){
        rev[x]=0; rev[l]^=1; rev[r]^=1;
        swap(lx[l],rx[l]); swap(lx[r],rx[r]);
        swap(son[l][0],son[l][1]); swap(son[r][0],son[r][1]);
    }
}
inline void rotate(int x){
    int y=fa[x],z=fa[y],k=is(x);
    son[z][is(y)]=x; fa[x]=z;
    son[y][k]=son[x][k^1]; fa[son[x][k^1]]=y;
    son[x][k^1]=y; fa[y]=x;
    pushup(y); pushup(x);
}
inline void splay(int x,int pos=0){
    while(fa[x]!=pos){
        int y=fa[x],z=fa[y];
        if(z!=pos) is(x)==is(y) ? rotate(y):rotate(x);
        rotate(x);
    }
    if(!pos) root=x;
}
inline int Find(int x,int k){
    pushdown(x);
    int l=son[x][0],r=son[x][1];
    if(sz[l]+1==k) return x;
    if(sz[l]+1>=k) return Find(l,k);
    else return Find(r,k-sz[l]-1);
}
inline void recycle(int x){
    if(son[x][0]) recycle(son[x][0]);
    if(son[x][1]) recycle(son[x][1]);
    que.push(x);
    son[x][0]=son[x][1]=tag[x]=rev[x]=fa[x]=0;
}
inline int split(int k,int len){
    int x=Find(root,k),y=Find(root,k+len+1);
    splay(x,0); splay(y,x);
    return son[son[root][1]][0];
}
inline int query(int k,int len){ return sum[split(k,len)]; }
inline void update(int k,int len,int v){
    int x=split(k,len),y=fa[x];
    tag[x]=1; val[x]=v; sum[x]=sz[x]*v;
    if(val[x]>=0) mx[x]=lx[x]=rx[x]=sum[x];
    else lx[x]=rx[x]=0,mx[x]=v;
    pushup(y); pushup(fa[y]);
}
inline void reverse(int k,int len){
    int x=split(k,len),y=fa[x];
    if(!tag[x]){
        rev[x]^=1;
        swap(son[x][0],son[x][1]); swap(lx[x],rx[x]);
        pushup(y); pushup(fa[y]);
    }
}
inline void del(int k,int len){
    int x=split(k,len),y=fa[x];
    recycle(x); son[y][0]=0;
    pushup(y); pushup(fa[y]);
}
inline void build(int rt,int l,int r){
    int mid=(l+r)>>1,now=id[mid],pre=id[rt];
    if(l==r){
        mx[now]=sum[now]=a[l];
        tag[now]=rev[now]=0; sz[now]=1;
        lx[now]=rx[now]=max(a[l],0);
    }
    if(l<mid) build(mid,l,mid-1);
    if(mid<r) build(mid,mid+1,r);
    val[now]=a[mid],fa[now]=pre;
    pushup(now); son[pre][mid>=rt]=now;
}
inline void Insert(int k,int len){
    for(int i=1;i<=len;i++) scanf("%d",&a[i]);
    for(int i=1;i<=len;i++){
        if(!que.empty()) id[i]=que.front(),que.pop();
        else id[i]=++cnt;
    }
    build(0,1,len);
    int z=id[(1+len)>>1],x=Find(root,k+1),y=Find(root,k+2);
    splay(x,0); splay(y,x);
    fa[z]=y; son[y][0]=z;
    pushup(y); pushup(x);
}
signed main(){
#ifndef ONLINE_JUDGE
    freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
#endif // ONLINE_JUDGE
    scanf("%d%d",&n,&m);
    a[1]=a[n+2]=mx[0]=-INF;
    for(int i=1;i<=n;i++) scanf("%d",&a[i+1]);
    for(int i=1;i<=n+2;i++) id[i]=i;
    build(0,1,n+2);
    root=(1+n+2)>>1; cnt=n+2;
    int k,len,v; char op[10];
    while(m--){
        scanf("%s",op);
        if(op[0]!='M' || op[2]!='X') scanf("%d%d",&k,&len);
        if(op[0]=='I') Insert(k,len);
        if(op[0]=='D') del(k,len);
        if(op[0]=='M')
            if(op[2]=='X') printf("%d\n",mx[root]);
            else scanf("%d",&v),update(k,len,v);
        if(op[0]=='R') reverse(k,len);
        if(op[0]=='G') printf("%d\n",query(k,len));
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值