NOI2005 维修数列 Splay

题目大意:
给出一个初始序列,要求实现下面的操作:
1.在当前序列的某个位置插入连续的若干个元素;
2.删除当前序列的某一段元素;
3.将当前序列中某一段元素全部修改为某个值;
4.翻转某个区间;
5.求当前序列某一段元素的总和;
6.求出当前序列中连续和的最大值。


一看就是Splay裸题,一看就知道操作很多很恶心。

各种操作当然要写准确了,主要注意下面几点:

1.显然有lazy操作,那么如何下放?何时下放

下放时可以参见线段树的处理方法,打上标记时,当前节点已经更新完毕。
由于几乎每个操作都要首先提取区间,而提取区间总要用到查找第k大数的函数。这个函数是自根向下进行的,这时遇到lazy标记显然必须下放。同时,找到编号后就会立即Splay,而这个点上跳时经过的点一定恰好是执行查找第k大函数时访问到的点,也就是说,这个时候该下放的一定都下放完毕了。
当然,为了保险,也可以在Splay操作的旋转前把节点到根路径上的点放在栈里面,先下放再旋转。

2.如何更新某个节点的各种值?

尤其注意Splay维护区间时更新与线段树有不同。线段树中,叶子节点的值是容易操作的,非叶子节点的的各种值仅由它的左右儿子的值决定,而且一定既有左儿子,又有右儿子
在Splay中,节点本身的各种值不仅受儿子影响,也有本身的影响。而且,非叶子节点有可能只有一个儿子。这个时候可以写四种情况逐个判断,也可以把编号0的节点(代表没有儿子)的值适当处理,使得不会对答案产生影响即可。

3.关于虚拟节点。

为了每次都能把想要的区间提取到根的右儿子的左子树上,分别在序列的首位添加一个占位元素。占位元素的Size必须是1,而其他值可以任取。处理操作6时,虽然已知讯问区间,还是要把整个区间提取出来,否则占位元素的值会对答案产生影响。(我就是这个操作没有处理对,调了2h+)

此外,插入操作和初始序列的建树可以采用与线段树类似的方法构建,这样不难实现,而且插入的是一棵严格的平衡树。


代码:

#include<stdio.h>
#include<algorithm>
#include<cstring>
#define MAXT 5000005
#define MAXN 500005
using namespace std;

int N,M;

int Tmax(int x,int y,int z)
{
    if(x>=y&&x>=z)return x;
    if(y>=z)return y;
    return z;
}

int rt,tot,fa[MAXT],ls[MAXT],rs[MAXT],Size[MAXT];
int val[MAXT],Sum[MAXT],MaxL[MAXT],MaxR[MAXT],MaxV[MAXT];
int lazy[MAXT],rev[MAXT];bool cl[MAXT];

void Update(int p)
{
    int l=ls[p],r=rs[p];

    Size[p]=Size[l]+Size[r]+1;
    Sum[p]=Sum[l]+Sum[r]+val[p];

    MaxL[p]=Tmax(MaxL[l],Sum[l]+val[p],Sum[l]+val[p]+MaxL[r]);
    MaxR[p]=Tmax(MaxR[r],Sum[r]+val[p],Sum[r]+val[p]+MaxR[l]);
    MaxV[p]=Tmax(MaxV[l],MaxV[r],Tmax(MaxR[l]+val[p],MaxL[r]+val[p],MaxR[l]+val[p]+MaxL[r]));
    if(MaxV[p]<val[p])MaxV[p]=val[p];
}

void Putdown(int p)
{
    if(cl[p]==false&&rev[p]==0)return;
    int l=ls[p],r=rs[p];
    if(rev[p])
    {
        rev[p]=0;rev[l]^=1;rev[r]^=1;
        swap(MaxL[l],MaxR[l]);swap(ls[l],rs[l]);
        swap(MaxL[r],MaxR[r]);swap(ls[r],rs[r]);
    }
    if(cl[p])
    {
        cl[p]=false;cl[l]=true;cl[r]=true;
        lazy[l]=lazy[r]=lazy[p];
        val[l]=val[r]=lazy[p];
        Sum[l]=Size[l]*lazy[p];
        Sum[r]=Size[r]*lazy[p];
        if(lazy[p]>0)
        {
            MaxL[l]=MaxR[l]=MaxV[l]=Size[l]*lazy[p];
            MaxL[r]=MaxR[r]=MaxV[r]=Size[r]*lazy[p];
        }
        else
        {
            MaxL[l]=MaxR[l]=MaxV[l]=lazy[p];
            MaxL[r]=MaxR[r]=MaxV[r]=lazy[p];
        }
    }
}

void Zig(int x)
{
    int y=fa[x],z=fa[y];
    if(z)
    {
        if(ls[z]==y)ls[z]=x;
        else rs[z]=x;
    }
    fa[x]=z;fa[y]=x;fa[rs[x]]=y;
    ls[y]=rs[x];rs[x]=y;
    Update(y);
    Update(x);
}

void Zag(int x)
{
    int y=fa[x],z=fa[y];
    if(z)
    {
        if(ls[z]==y)ls[z]=x;
        else rs[z]=x;
    }
    fa[x]=z;fa[y]=x;fa[ls[x]]=y;
    rs[y]=ls[x];ls[x]=y;
    Update(y);
    Update(x);
}

void Splay(int x,int t)
{
    int y,z;
    while(fa[x]!=t)
    {
        y=fa[x];z=fa[y];
        if(z==t)
        {
            if(ls[y]==x)Zig(x);
            else Zag(x);
        }
        else
        {
            if(ls[z]==y){
                if(ls[y]==x)Zig(y),Zig(x);
                else Zag(x),Zig(x);
            }
            else
            {
                if(rs[y]==x)Zag(y),Zag(x);
                else Zig(x),Zag(x);
            }
        }
    }
    if(!t)rt=x;
}

int A[MAXN];
int Build(int x,int y,int f)
{
    int p=++tot,mid=x+y>>1;
    fa[p]=f;val[p]=A[mid];
    if(x==y)
    {
        Size[p]=1;
        Sum[p]=MaxL[p]=MaxR[p]=MaxV[p]=A[mid];
        return p;
    }
    if(x<=mid-1)ls[p]=Build(x,mid-1,p);
    if(mid+1<=y)rs[p]=Build(mid+1,y,p);
    Update(p);
    return p;
}

int GetKth(int k)
{
    int p=rt;
    while(p)
    {
        Putdown(p);
        if(Size[ls[p]]+1==k)return p;
        if(Size[ls[p]]>=k)p=ls[p];
        else k-=Size[ls[p]]+1,p=rs[p];
    }
}

void Extract(int l,int r)//将区间[l,r]提取到根的右儿子的左子树上
{
    int p,q;
    p=GetKth(l);
    Splay(p,0);
    q=GetKth(r+2);
    Splay(q,rt);
}

void Ins()
{
    int pos,sz,i,p,q;

    scanf("%d%d",&pos,&sz);
    for(i=1;i<=sz;i++)scanf("%d",&A[i]);
    p=GetKth(pos+1);Splay(p,0);
    q=GetKth(pos+2);Splay(q,rt);
    ls[rs[rt]]=Build(1,sz,rs[rt]);
    Update(rs[rt]);
    Update(rt);
}

void Del(int l,int r)
{
    Extract(l,r);
    int p=ls[rs[rt]];
    fa[p]=0;ls[rs[rt]]=0;
    Update(rs[rt]);
    Update(rt);
}

void Modify(int l,int r,int k)
{
    Extract(l,r);
    int p=ls[rs[rt]];

    lazy[p]=k;cl[p]=true;
    val[p]=k;
    Sum[p]=Size[p]*k;
    if(k>0)MaxL[p]=MaxR[p]=MaxV[p]=Size[p]*k;
    else MaxL[p]=MaxR[p]=MaxV[p]=k;

    Update(rs[rt]);
    Update(rt);
}

void Reverse(int l,int r)
{
    Extract(l,r);
    int p=ls[rs[rt]];
    rev[p]^=1;
    swap(MaxL[p],MaxR[p]);
    swap(ls[p],rs[p]);
}

int GetSum(int l,int r)
{
    Extract(l,r);
    return Sum[ls[rs[rt]]];
}


int main()
{
    Sum[0]=0;
    MaxL[0]=MaxR[0]=MaxV[0]=val[0]=-1e9;
    Size[0]=0;
    //上面三行就是“对0号点的适当处理”

    int i,pos,sz,c;
    char op[20];

    scanf("%d%d",&N,&M);
    for(i=1;i<=N;i++)scanf("%d",&A[i+1]);

    rt=Build(1,N+2,0);
    for(i=1;i<=M;i++)
    {
        scanf("%s",op);
        if(op[0]=='I')Ins();
        else if(op[0]=='D')
        {
            scanf("%d%d",&pos,&sz);
            Del(pos,pos+sz-1);
        }
        else if(op[0]=='R')
        {
            scanf("%d%d",&pos,&sz);
            Reverse(pos,pos+sz-1);
        }
        else if(op[0]=='G')
        {
            scanf("%d%d",&pos,&sz);
            printf("%d\n",GetSum(pos,pos+sz-1));
        }
        else if(op[2]=='K') 
        {
            scanf("%d%d%d",&pos,&sz,&c);
            Modify(pos,pos+sz-1,c);
        }
        else
        {
            Extract(1,Size[rt]-2);
            printf("%d\n",MaxV[ls[rs[rt]]]);
        }
    }
}

此外,Update函数也可以用分类讨论写,只不过略恶心:

void Update(int p)
{
    int l=ls[p],r=rs[p];

    if(l)
    {
        if(r)
        {
            Size[p]=Size[l]+Size[r]+1;
            Sum[p]=Sum[l]+Sum[r]+val[p];
            MaxL[p]=Tmax(MaxL[l],Sum[l]+val[p],Sum[l]+val[p]+MaxL[r]);
            MaxR[p]=Tmax(MaxR[r],Sum[r]+val[p],Sum[r]+val[p]+MaxR[l]);
            MaxV[p]=Tmax(MaxV[l],MaxV[r],Tmax(val[p]+MaxR[l],val[p]+MaxL[r],MaxR[l]+val[p]+MaxL[r]));
            MaxV[p]=max(MaxV[p],val[p]);
        }
        else
        {
            Size[p]=Size[l]+1;
            Sum[p]=Sum[l]+val[p];
            MaxL[p]=max(MaxL[l],Sum[l]+val[p]);
            MaxR[p]=max(val[p],MaxR[l]+val[p]);
            MaxV[p]=Tmax(val[p],MaxV[l],MaxR[l]+val[p]);
        }
    }
    else
    {
        if(r)
        {
            Size[p]=Size[r]+1;
            Sum[p]=Sum[r]+val[p];
            MaxL[p]=max(val[p],val[p]+MaxL[r]);
            MaxR[p]=max(MaxR[r],Sum[r]+val[p]);
            MaxV[p]=Tmax(val[p],MaxV[r],MaxL[r]+val[p]);
        }
        else
        {
            Size[p]=1;
            Sum[p]=val[p];
            MaxL[p]=MaxR[p]=MaxV[p]=val[p];
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值