【介绍】
伸展树(Splay Tree),也叫分裂树,是一种二叉排序树,它能在O(log n)内完成插入、查找和删除操作。
【操作】
Ps:
struct Splay
{
Splay* son[2];
int num,s,sum,pre,suf,MAX,Save; //num表示点权,s表示这个子树的点数,sum表示子树的点权加和,pre表示子树前缀的最大值,suf表示子树后缀的最大值,MAX表示子树的子段的最大值,Save是lazy-tag。
bool flip; // flip是lazy-tag。
int Cmp(int &k){if (k==son[0]->s+1) return -1; if (k<son[0]->s+1) return 0; k-=son[0]->s+1; return 1;}
void Pushup(){
s=son[0]->s+son[1]->s+1; sum=son[0]->sum+son[1]->sum+num;
pre=max(son[0]->pre,max(son[0]->sum+num,son[0]->sum+num+son[1]->pre));
suf=max(son[1]->suf,max(son[1]->sum+num,son[1]->sum+num+son[0]->suf));
MAX=max(son[0]->MAX,son[1]->MAX);
MAX=max(MAX,max(max(0,son[0]->suf+son[1]->pre),max(son[0]->suf,son[1]->pre))+num); //以上是修正pre,suf,MAX,根据定义应该不难懂。
}
}tem[maxn],*Null=tem,*ro=Null,*stk[maxn];//std是存储暂时没用到的指针。
void Pushflip(Splay* k)//交换左右儿子
{
swap(k->son[0],k->son[1]); swap(k->pre,k->suf); k->flip^=1;
}
void PushSave(Splay* k,int num) //修改一系列数值
{
k->num=k->Save=num; k->sum=k->s*num;
if (num>0) k->MAX=k->pre=k->suf=k->s*num; else k->MAX=k->pre=k->suf=num;
}
void Pushdown(Splay* k)//传递lazy-tag
{
if (k->flip)
{
if (k->son[0]!=Null) Pushflip(k->son[0]);
if (k->son[1]!=Null) Pushflip(k->son[1]);
k->flip=0;
}
if (k->Save!=-INF)
{
if (k->son[0]!=Null) PushSave(k->son[0],k->Save);
if (k->son[1]!=Null) PushSave(k->son[1],k->Save);
k->Save=-INF;
}
}
void New(Splay* &k,int num)//新建节点。
{
k=stk[top--]; k->son[0]=k->son[1]=Null; k->flip=0; k->Save=-INF; k->s=1; k->num=k->sum=k->pre=k->suf=k->MAX=num;
}
① 伸展:
伸展操作是把一个节点伸展到根上的操作,且需要分不同状况伸展。
(1)将根节点的某一个儿子旋转到根节点上
(2)”三点共线”时,整棵树往同一方向旋转两次。
(3)”三点不共线”时,先旋转子树,再旋转。
void Rotate(Splay* &k,int d)//旋转
{
Splay* t=k->son[d]; k->son[d]=t->son[d^1]; t->son[d^1]=k;
k->Pushup(); t->Pushup(); k=t;
}
void Splay_Work(Splay* &k,int x)//伸展
{
Pushdown(k);
int d1=k->Cmp(x);
if (d1>-1)
{
Splay* p=k->son[d1]; Pushdown(p);
int d2=p->Cmp(x);
if (d2>-1)
{
Splay_Work(p->son[d2],x);
if (d1==d2) Rotate(k,d1),Rotate(k,d1); else Rotate(k->son[d1],d2),Rotate(k,d1);
} else Rotate(k,d1);
}
}
② 建树:
Splay建树就是将一个序列转成一棵树。
Splay* Build(int L,int R)//建树
{
if (L>R) return Null;
int mid=(R-L>>1)+L;
Splay* now; New(now,a[mid]);
now->son[0]=Build(L,mid-1); now->son[1]=Build(mid+1,R);
now->Pushup();
return now;
}
③ 定位:
将L-1伸展到根节点并将R+1伸展到根节点的右儿子,这样L~R就定位在以根节点的右儿子的左儿子的子树上了。
void GetLR(Splay* &k,int L,int R) //定位
{
L--; R++; Splay_Work(k,L);
k->Cmp(R); Splay_Work(k->son[1],R);
}
④ 插入:
定位好位置后,将新的树插入到根节点的右儿子的左儿子上。
void Insert(Splay* &k,int where,Splay* now)
{
GetLR(k,where+1,where);
k->son[1]->son[0]=now;
ro->son[1]->Pushup(); ro->Pushup();
}
⑤ 删除:
定位好位置后,将以根节点的右儿子的左儿子的子树删除即删除L~R这段序列。
void Delete_Work(Splay* k)
{
if (k==Null) return;
Delete_Work(k->son[0]); stk[++top]=k; Delete_Work(k->son[1]);//回收没用的指针。
}
void Delete(Splay* &k,int L,int R)//删除
{
GetLR(k,L,R);
Delete_Work(k->son[1]->son[0]);
k->son[1]->son[0]=Null;
ro->son[1]->Pushup(); ro->Pushup();
}