【题解】洛谷P2042[NOI2005]维护数列 splay
之前大佬的模板原来就是这题的啊……又学一遍（还是不懂）
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=5e5+10,INF=0x3f3f3f3f;
int pre[N],ch[N][2],key[N],size[N],root,tot1,n,m;
int sum[N],rev[N],same[N],lx[N],rx[N],mx[N];
int s[N],tot2,a[N];
void newnode(int &r,int fa,int k)
{
if(tot2)r=s[tot2--];
else r=++tot1;
pre[r]=fa;ch[r][0]=ch[r][1]=0;
key[r]=sum[r]=k;
rev[r]=same[r]=0;
lx[r]=rx[r]=mx[r]=k;
size[r]=1;
return;
}
void push_up(int r)
{
int lson=ch[r][0],rson=ch[r][1];
size[r]=size[lson]+size[rson]+1;
sum[r]=sum[lson]+sum[rson]+key[r];
lx[r]=max(lx[lson],sum[lson]+key[r]+max(0,lx[rson]));
rx[r]=max(rx[rson],sum[rson]+key[r]+max(0,rx[lson]));
mx[r]=max(0,rx[lson])+key[r]+max(0,lx[rson]);
mx[r]=max(mx[r],max(mx[lson],mx[rson]));
return;
}
void build(int &x,int l,int r,int fa)
{
if(l>r)return;
int mid=l+r>>1;
newnode(x,fa,a[mid]);
build(ch[x][0],l,mid-1,x);
build(ch[x][1],mid+1,r,x);
push_up(x);
return;
}
//把权值修改为v
void updatesame(int r,int v)
{
if(!r)return;
key[r]=v;sum[r]=v*size[r];
lx[r]=rx[r]=mx[r]=max(v,v*size[r]);
same[r]=1;
return;
}
void updaterev(int r)
{
if(!r)return;
swap(ch[r][0],ch[r][1]);
swap(lx[r],rx[r]);
rev[r]^=1;
return;
}
//下传延迟标记
void push_down(int r)
{
if(same[r])
{
updatesame(ch[r][0],key[r]);
updatesame(ch[r][1],key[r]);
same[r]=0;
}
if(rev[r])
{
updaterev(ch[r][0]);
updaterev(ch[r][1]);
rev[r]=0;
}
return;
}
void Init()
{
root=tot1=tot2=0;
ch[root][0]=ch[root][1]=size[root]=pre[root]=same[root]=0;
rev[root]=sum[root]=key[root]=0;
lx[root]=rx[root]=mx[root]=-INF;
newnode(root,0,-1);newnode(ch[root][1],root,-1);
for(int i=0;i<n;i++)scanf("%d",&a[i]);
build(ch[ch[root][1]][0],0,n-1,ch[root][1]);
push_up(ch[root][1]);push_up(root);
return;
}
//得到第k个标号
int getk(int r,int k)
{
push_down(r);
int t=size[ch[r][0]]+1;
if(t==k)return r;
if(t>k)return getk(ch[r][0],k);
else return getk(ch[r][1],k-t);
}
//旋转，0为左旋，1为右旋
void Rotate(int x,int kind)
{
int y=pre[x];
push_down(y);push_down(x);
ch[y][!kind]=ch[x][kind];
pre[ch[x][kind]]=y;
if(pre[y])ch[pre[y]][ch[pre[y]][1]==y]=x;
pre[x]=pre[y];
ch[x][kind]=y;
pre[y]=x;
push_up(y);
return;
}
//将r结点调整到goal下面
void splay(int r,int goal)
{
push_down(r);
while(pre[r]!=goal)
{
if(pre[pre[r]]==goal)
{
push_down(pre[r]);
push_down(r);
Rotate(r,ch[pre[r]][0]==r);
}
else
{
push_down(pre[pre[r]]);
push_down(pre[r]);
push_down(r);
int y=pre[r];
int kind=ch[pre[y]][0]==y;
if(ch[y][kind]==r)
{
Rotate(r,!kind);
Rotate(r,kind);
}
else
{
Rotate(y,kind);
Rotate(r,kind);
}
}
}
push_up(r);
if(goal==0)root=r;
return;
}
//返回最大子段和
int getmaxsum(int pos,int tot)
{
splay(getk(root,pos),0);
splay(getk(root,pos+tot+1),root);
return mx[ch[ch[root][1]][0]];
}
//从第pos个数开始的连续tot个数修改为c
void makesame(int pos,int tot,int c)
{
splay(getk(root,pos),0);
splay(getk(root,pos+tot+1),root);
updatesame(ch[ch[root][1]][0],c);
push_up(ch[root][1]);
push_up(root);
return;
}
//在第pos个数后面插入tot个数
void Insert(int pos,int tot)
{
for(int i=0;i<tot;i++)scanf("%d",&a[i]);
splay(getk(root,pos+1),0);
splay(getk(root,pos+2),root);
build(ch[ch[root][1]][0],0,tot-1,ch[root][1]);
push_up(ch[root][1]);
push_up(root);
return;
}
//删除子树
void Erase(int r)
{
if(!r)return;
s[++tot2]=r;
Erase(ch[r][0]);
Erase(ch[r][1]);
return;
}
//从第pos个数开始删去tot个数
void Delete(int pos,int tot)
{
splay(getk(root,pos),0);
splay(getk(root,pos+tot+1),root);
Erase(ch[ch[root][1]][0]);
pre[ch[ch[root][1]][0]]=0;
ch[ch[root][1]][0]=0;
push_up(ch[root][1]);
push_up(root);
return;
}
//将第pos个数开始的连续tot个数翻转
void Reverse(int pos,int tot)
{
splay(getk(root,pos),0);
splay(getk(root,pos+tot+1),root);
updaterev(ch[ch[root][1]][0]);
push_up(ch[root][1]);
push_up(root);
return;
}
//得到第pos个数开始连续tot个数的和
int getsum(int pos,int tot)
{
splay(getk(root,pos),0);
splay(getk(root,pos+tot+1),root);
return sum[ch[ch[root][1]][0]];
}
int main()
{
//freopen("in.txt","r",stdin);
scanf("%d%d",&n,&m);
Init();
char op[20];int pos,tot;
while(m--)
{
scanf("%s", op);
if(strcmp(op,"INSERT")==0)
{
scanf("%d%d",&pos,&tot);
Insert(pos,tot);
}
else if(strcmp(op,"DELETE")==0)
{
scanf("%d%d",&pos,&tot);
Delete(pos,tot);
}
else if(strcmp(op,"MAKE-SAME")==0)
{
int c;
scanf("%d%d%d",&pos,&tot,&c);
makesame(pos,tot,c);
}
else if(strcmp(op,"REVERSE")==0)
{
scanf("%d%d",&pos,&tot);
Reverse(pos,tot);
}
else if(strcmp(op,"GET-SUM")==0)
{
scanf("%d%d",&pos,&tot);
printf("%d\n",getsum(pos,tot));
}
else if(strcmp(op,"MAX-SUM")==0)
{
printf("%d\n",getmaxsum(1,size[root]-2));
}
}
return 0;
}
总结
splay的画风……码农的节奏