题意:HYSBZ 1500 维修数列(伸展树模板)-LMLPHP

题解:典型伸展树的题,比较全面。

我理解的伸展树:

1 伸展操作:就是旋转,因为我们只需保证二叉树中序遍历的结果不变,所以我们可以旋转来保持树的平衡,且旋转有左旋与右旋。通过这种方式保证不会让树一直退化从而超时。虽然一次旋转的代价比较高,但是可以证明:每次操作都旋转(关键),则时间复杂度为O(n*log2 n)

2 更新:每个节点都可以存一些信息,并模拟线段树进行区间操作。父节点的信息是两个孩子节点加当前父节点的信息的总和。因为是可旋转的搜索二叉树,所以每次处理都需要注意上更新或下更新

3 注意:一般需要先开两个哨兵节点,一个作为开头,有个作为结尾,这样可以避免一些边界的讨论问题

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define dir(a,b) (a>>b)
#define ssplay(rt,x) (splay[rt].chd[x])
typedef long long ll;
const int Max=6e5+;
const int Inf=<<;
int num[Max],memp[Max],tot2,tot,root;//存值 内存池 内存池的值 总值 根节点
struct node
{
int file,sam;//翻转 是否修改
int chd[],fat;
int sizee,sum,val;//总个数 总大小 值
int lmax,rmax,mmax;//区间合并三变量
} splay[Max]; void Treaval(int rt)
{
if(rt)
{
Treaval(splay[rt].chd[]);
printf("rt=%2d lchd=%2d rchd=%2d sum=%2d size=%2d val=%2d lmax=%2d rmax=%2d mmax=%2d\n",rt,splay[rt].chd[],splay[rt].chd[],
splay[rt].sum,splay[rt].sizee,splay[rt].val,splay[rt].lmax,splay[rt].rmax,splay[rt].mmax);
Treaval(splay[rt].chd[]);
}
return;
}
void debug()
{
printf("root=%d\n",root);
Treaval(root);
return;
} inline void NewNode(int &rt,int fa,int va)//建立新节点
{
if(tot2)//删除后的内存池可以再次利用
rt=memp[tot2--];
else
rt=++tot;
splay[rt].file=splay[rt].sam=;
splay[rt].chd[]=splay[rt].chd[]=;
splay[rt].val=splay[rt].lmax=splay[rt].rmax=splay[rt].mmax=splay[rt].sum=va;
splay[rt].fat=fa;
return;
} inline int nmax(int a,int b)
{
return a>b?a:b;
}
inline void PushUp(int rt)//上更新(类似区间合并)
{
int lchd=splay[rt].chd[],rchd=splay[rt].chd[];
splay[rt].sizee=splay[lchd].sizee+splay[rchd].sizee+;
splay[rt].sum=splay[lchd].sum+splay[rchd].sum+splay[rt].val; splay[rt].mmax=nmax(nmax(splay[rt].val,splay[lchd].mmax),splay[rchd].mmax);//处理区间最大值
if(splay[rt].mmax>)
splay[rt].mmax=nmax(splay[rt].mmax,nmax(splay[lchd].rmax,)+nmax(splay[rchd].lmax,)+splay[rt].val);//val为关键 splay[rt].lmax=nmax(splay[lchd].lmax,splay[lchd].sum+splay[rt].val);
splay[rt].lmax=nmax(splay[rt].lmax,splay[lchd].sum+splay[rt].val+splay[rchd].lmax); splay[rt].rmax=nmax(splay[rchd].rmax,splay[rchd].sum+splay[rt].val);
splay[rt].rmax=nmax(splay[rt].rmax,splay[rchd].sum+splay[rt].val+splay[lchd].rmax);
return;
}
inline void Swap(int &a,int &b)
{
int t=a;
a=b;
b=t;
return;
}
inline void fson(int rt)//翻转
{
if(!rt)
return;
Swap(splay[rt].chd[],splay[rt].chd[]);//孩子交换就好
Swap(splay[rt].lmax,splay[rt].rmax);//此位置的左右max需交换
splay[rt].file^=;//此处修改与否只与父节点flie有关,与此处的file无关
}
inline void sson(int rt,int va)//修改成va
{
if(!rt)
return;
splay[rt].val=va;
splay[rt].sum=va*splay[rt].sizee;
splay[rt].lmax=splay[rt].rmax=splay[rt].mmax=nmax(splay[rt].sum,va);
splay[rt].sam=;
}
inline void PushDown(int rt)//下更新(处理翻转与改变值)
{
if(splay[rt].file)
{
fson(splay[rt].chd[]);
fson(splay[rt].chd[]);
splay[rt].file=;
}
if(splay[rt].sam)
{
sson(splay[rt].chd[],splay[rt].val);
sson(splay[rt].chd[],splay[rt].val);
splay[rt].sam=;
}
return;
} inline void Rotate(int rt,int kind)//**zig或者zag**
{
int y=splay[rt].fat;
PushDown(y);
PushDown(rt);
splay[y].chd[kind^]=splay[rt].chd[kind];
splay[ssplay(rt,kind)].fat=y;
if(splay[y].fat)//不是一个zig后者zag
splay[splay[y].fat].chd[ssplay(splay[y].fat,)==y]=rt;//y父节点的(y的左右)孩子
splay[rt].fat=splay[y].fat;
splay[rt].chd[kind]=y;
splay[y].fat=rt;
PushUp(y);
return;
}
inline void Splay(int rt,int goal)//**关键的伸展操作(双旋)**
{
PushDown(rt);
while(splay[rt].fat!=goal)
{
int y=splay[rt].fat;
if(splay[y].fat==goal)//一次zig/zag
{
Rotate(rt,splay[y].chd[]==rt);//rt是否为左孩子
}
else
{
int kind=(splay[splay[y].fat].chd[]==y?:);//y是否为左孩子
if(splay[y].chd[kind]==rt)//左孩子的右孩子或者右孩子的左孩子
{
Rotate(rt,kind^);
Rotate(rt,kind);
}
else
{
Rotate(y,kind);
Rotate(rt,kind);
}
}
}
PushUp(rt);
if(!goal)
root=rt;//更新根节点
return;
}
inline void Rotateto(int pos,int goal)//**得到第pos个数,并且进行伸展**
{
int rt=root;
PushDown(rt);
while(splay[ssplay(rt,)].sizee!=pos)
{
if(splay[ssplay(rt,)].sizee>pos)
rt=splay[rt].chd[];
else
{
pos-=(splay[ssplay(rt,)].sizee+);
rt=splay[rt].chd[];
}
PushDown(rt);
}
Splay(rt,goal);
return;
} void Create(int sta,int enn,int &rt,int fa)//建树与添树
{
if(sta>enn)
return;
int mid=dir(sta+enn,);
NewNode(rt,fa,num[mid]);
Create(sta,mid-,splay[rt].chd[],rt);
Create(mid+,enn,splay[rt].chd[],rt);
PushUp(rt);//建树与添树时上更新
return;
}
void Init(int n)//初始化
{
for(int i=; i<n; ++i)
scanf("%d",&num[i]);
splay[].lmax=splay[].rmax=splay[].mmax=-Inf;//可能全为负数
splay[].chd[]=splay[].chd[]=splay[].fat=;//建立哨兵,避免特判
splay[].val=splay[].sizee=splay[].sum=tot2=tot=root=;
splay[].sam=splay[].file=;
NewNode(root,,);//建立两个哨兵
NewNode(splay[root].chd[],root,);
Create(,n-,splay[ssplay(root,)].chd[],splay[root].chd[]);
PushUp(splay[root].chd[]);//与建树一起的上更新
PushUp(root);
return;
}
void Insert(int pos,int dig)//在pos与pos+1之间添加
{
Rotateto(pos,);//旋转pos位置的值成0的孩子
Rotateto(pos+,root);
Create(,dig-,splay[ssplay(root,)].chd[],splay[root].chd[]);//建立
PushUp(splay[root].chd[]);//更新
PushUp(root);//下面很多函数都是五行中仅仅修改第三行
return;
} void Erase(int rt)//回收空间
{
if(!rt)
return;
memp[++tot2]=rt;
Erase(splay[rt].chd[]);
Erase(splay[rt].chd[]);
return;
}
void Delete(int pos,int dig)//删除pos后面dig个
{
Rotateto(pos-,);
Rotateto(pos+dig,root);
Erase(splay[ssplay(root,)].chd[]);//关键位置
splay[ssplay(root,)].chd[]=;
PushUp(splay[root].chd[]);
PushUp(root);
return;
} void Make_same(int pos,int dig,int fix)//修改pos后面dig和为fix
{
Rotateto(pos-,);
Rotateto(pos+dig,root);
sson(splay[ssplay(root,)].chd[],fix);
PushUp(splay[root].chd[]);
PushUp(root);
return;
} void Reverse(int pos,int dig)//翻转pos后面dig个
{
Rotateto(pos-,);
Rotateto(pos+dig,root);
fson(splay[ssplay(root,)].chd[]);
PushUp(splay[root].chd[]);
PushUp(root);
return;
} int GetSum(int pos,int dig)//计算pos后面dig个数的和
{
Rotateto(pos-,);
Rotateto(pos+dig,root);
return splay[ssplay(ssplay(root,),)].sum;
} int GetMaxsum(int pos,int dig)//区间最值
{
Rotateto(pos-,);
Rotateto(pos+dig,root);
return splay[ssplay(ssplay(root,),)].mmax;//注意每次都需要splay一下,保证时间复杂度
} int main()
{
int n,m;
char str[];
int pos,dig,fix;
while(~scanf("%d %d",&n,&m))
{
Init(n);
while(m--)
{
//debug();
scanf("%s",str);
if(!strcmp(str,"INSERT"))//添加一段数
{
scanf("%d %d",&pos,&dig);
for(int i=; i<dig; ++i)
scanf("%d",&num[i]);
Insert(pos,dig);
}
else if(!strcmp(str,"DELETE"))//删除一段数
{
scanf("%d %d",&pos,&dig);
Delete(pos,dig);
}
else if(!strcmp(str,"MAKE-SAME"))//修改
{
scanf("%d %d %d",&pos,&dig,&fix);
Make_same(pos,dig,fix);
}
else if(!strcmp(str,"REVERSE"))//翻转
{
scanf("%d %d",&pos,&dig);
Reverse(pos,dig);
}
else if(!strcmp(str,"GET-SUM"))//求和
{
scanf("%d %d",&pos,&dig);
printf("%d\n",GetSum(pos,dig));
}
else//最大子序列
{
printf("%d\n",GetMaxsum(,splay[root].sizee-));//注意有两个哨兵
}
}
}
return ;
}
05-11 20:06