题目大意:
给出一个初始序列,要求实现下面的操作:
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];
}
}
}