- 今天刚刚学习完splay,讲一下自己的想法吧
- 首先splay和treap不一样,treap通过随机数来调整树的形态。但splay不一样,再每插入或操作一次后,你都会把他旋转到根,再旋转过程中树的形态会不断改变,这样来达到均摊效果
常数据大。 - 来看看具体实现吧
首先定义数组,\(size\) 子树大小(包括自己),\(fa\) 节点的父亲,\(key\) 该节点的权值,\(cnt\) 该节点权值出现次数,$ch $表示儿子 0表左二子,1表右儿子
首先看几个简单函数
inline void update(int x)
{
size[x]=cnt[x]+size[ch[x][0]]+size[ch[x][1]];
}
更新子树大小
inline int get(int x){return x==ch[fa[x]][1];}
返回该节点是left儿子还是right儿子
inline void clear(int x){ch[x][0]=ch[x][1]=fa[x]=size[x]=cnt[x]=key[x]=0;}
删除该节点,清空所有信息
接下来是splay的精髓所在
inline void rotate(int x,int &k)
{
static int old,oldfa,o;
old=fa[x];oldfa=fa[old];o=get(x);
if(old==k)k=x;
else ch[oldfa][get(old)]=x;
fa[x]=oldfa;
ch[old][o]=ch[x][o^1];fa[ch[x][o^1]]=old;
ch[x][o^1]=old;fa[old]=x;
update(x),update(old);
}
inline void splay(int x,int &k)
{
while(x!=k)
{
if(fa[x]!=k)rotate(get(x)^get(fa[x])?x:fa[x],k);
rotate(x,k);
}
}
rotate,splay,是splay核心操作,显然splay是依赖于rotate的,让我们看一下rotate是如何实现的吧
(手绘图)
我们考虑从图上左往右的过程,我们要将y旋上去,因为y本是x的右儿子,所以x放到y的左儿子,将y的原本左儿子设为x的右儿子,这是左旋,还有对称操作右旋,但我们不必要打两个函数,用 ^可以实现左右儿子的转换,用get操作实现,具体实现参考代码,打代码时最好画个图参照一下。
splay,这个操作完全依靠rotate,目的就是把你要的节点旋转到k(一般是root),k要传地址,要修改。在while循环里加了个小小的优化,但x和他的fa在同一侧时可以旋fa,小小的加速
inline void insert(int x)
{
if(!root){root=++sz;size[sz]=cnt[sz]=1;key[sz]=x;return;}
int now=root,o;
while(1)
{
if(x==key[now])
{
++cnt[now];
splay(now,root);
update(now);
return;
}
o=x>key[now]?1:0;
if(!ch[now][o])
{
ch[now][o]=++sz;
size[sz]=cnt[sz]=1;
key[sz]=x;fa[sz]=now;
ch[now][o]=sz;
update(now);
splay(sz,root);
return;
}
else now=ch[now][o];
}
}
insert,插入一个数,当没有数时就直接把这个数设为根,else 因为树满足二叉排序树的性质,所以比当前节点的key小就往左走,否则往右走,直到找到一个空节点,更新信息,由于这个点以上所有的点\(size\)都要加一,不好update,所以把这给点旋转到根,将这个点update就行了
inline int find_pos(int x)
{
int now=root;
while(1)
{
if(x==key[now]){return now;}
if(x<key[now])now=ch[now][0];
else now=ch[now][1];
}
}
找到该值在树中的节点编号
inline int pre()
{
int now=ch[root][0];
while(ch[now][1])now=ch[now][1];
return now;
}
inline int nex()
{
int now=ch[root][1];
while(ch[now][0])now=ch[now][0];
return now;
}
求前驱,后继,前驱从根的左儿子开始一直往右跑,后继从根的右儿子开始一直往左跑即可
void del(int x)
{
splay(find_pos(x),root);
if(cnt[root]>1){--cnt[root];return;}
if(!ch[root][0]&&!ch[root][1]){clear(root);root=0;return;}
if(ch[root][0]&&ch[root][1])
{
int oldroot=root;
splay(pre(),root);
fa[ch[oldroot][1]]=root;
ch[root][1]=ch[oldroot][1];
clear(oldroot);
update(root);
}
else
{
int o=ch[root][1]>0;
root=ch[root][o];
clear(fa[root]);
fa[root]=0;
}
}
删除操作,有点麻烦,先找到x的位置
- 如果x有多个就\(cnt\)减一
- 如果一个儿子都没有就直接删掉,root设为0
- 如果 只有一个儿子就把儿子设为根,删去这个点
- 剩下两个儿子情况,找到根的前驱,把前驱旋到根,这是root只有左儿子,再把原来根的右儿子到root上,这样原来的root就脱离了树,再删掉即可。
inline int find_order_of_key(int x)
{
int res=0,now=root;
while(1)
{
if(x<key[now])now=ch[now][0];
else
{
res+=size[ch[now][0]];
if(x==key[now]){splay(now,root);return res+1;}
res+=cnt[now];
now=ch[now][1];
}
}
}
inline int find_by_order(int x)
{
int now=root;
while(1)
{
if(x<=size[ch[now][0]])now=ch[now][0];
else
{
int temp=size[ch[now][0]]+cnt[now];
if(x<=temp)return key[now];
else{x-=temp;now=ch[now][1];}
}
}
}
找x的排名,与找排名为x的数,其实大同小异,用二叉搜索树的性质即可,只是记得答案不一样罢了
inline void rever(int l,int r)
{
l=find(l-1);r=find(r+1);
splay(l,root);splay(r,ch[l][1]);
rev[ch[r][0]]^=1;
}
找到区间左边一个和区间的右边一个点在树中位置,把左边的点旋转到根,再把右边的点旋到root的右儿子,这时这段区间一定是ch[r][0]的子树(想一想,为什么)(根据二叉搜索树的性质),把这个点打上标记即可;当遇到有翻转标记的点时,交换其左右子树,并下传标记即可。
ok,splay的基本操作就是这些了
下面是完整代码
洛谷P3369 treap模板
#include<bits/stdc++.h>
using namespace std;
typedef int sign;
typedef long long ll;
#define For(i,a,b) for(register sign i=(sign)a;i<=(sign)b;++i)
#define Fordown(i,a,b) for(register sign i=(sign)a;i>=(sign)b;--i)
const int N=1e5+5;
void cmax(sign &a,sign b){if(a<b)a=b;}
void cmin(sign &a,sign b){if(a>b)a=b;}
template<typename T>T read()
{
T ans=0,f=1;
char ch=getchar();
while(!isdigit(ch)&&ch!=‘-‘)ch=getchar();
if(ch==‘-‘)f=-1,ch=getchar();
while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch-‘0‘),ch=getchar();
return ans*f;
}
void file()
{
#ifndef ONLINE_JUDGE
freopen("splay.in","r",stdin);
freopen("splay.out","w",stdout);
#endif
}
int fa[N],size[N],key[N],cnt[N],ch[N][2],sz,root;
inline void update(int x){size[x]=cnt[x]+size[ch[x][0]]+size[ch[x][1]];}
inline int get(int x){return x==ch[fa[x]][1];}
inline void clear(int x){ch[x][0]=ch[x][1]=fa[x]=size[x]=cnt[x]=key[x]=0;}
inline void rotate(int x,int &k)
{
static int old,oldfa,o;
old=fa[x];oldfa=fa[old];o=get(x);
if(old==k)k=x;
else ch[oldfa][get(old)]=x;
fa[x]=oldfa;
ch[old][o]=ch[x][o^1];fa[ch[x][o^1]]=old;
ch[x][o^1]=old;fa[old]=x;
update(x),update(old);
}
inline void splay(int x,int &k)
{
while(x!=k)
{
if(fa[x]!=k)rotate(get(x)^get(fa[x])?x:fa[x],k);
rotate(x,k);
}
}
inline void insert(int x)
{
//puts("");
if(!root){root=++sz;size[sz]=cnt[sz]=1;key[sz]=x;return;}
int now=root,o;
while(1)
{
if(x==key[now])
{
++cnt[now];
splay(now,root);
update(now);
return;
}
o=x>key[now]?1:0;
if(!ch[now][o])
{
ch[now][o]=++sz;
size[sz]=cnt[sz]=1;
key[sz]=x;fa[sz]=now;
ch[now][o]=sz;
update(now);
splay(sz,root);
return;
}
else now=ch[now][o];
//printf("%d %d %d %d\n",now,fa[now],ch[now][0],ch[now][1]);
}
}
inline int find_pos(int x)
{
int now=root;
while(1)
{
if(x==key[now]){return now;}
if(x<key[now])now=ch[now][0];
else now=ch[now][1];
}
}
inline int pre()
{
int now=ch[root][0];
while(ch[now][1])now=ch[now][1];
return now;
}
inline int nex()
{
int now=ch[root][1];
while(ch[now][0])now=ch[now][0];
return now;
}
void del(int x)
{
splay(find_pos(x),root);
if(cnt[root]>1){--cnt[root];return;}
if(!ch[root][0]&&!ch[root][1]){clear(root);root=0;return;}
if(ch[root][0]&&ch[root][1])
{
int oldroot=root;
splay(pre(),root);
fa[ch[oldroot][1]]=root;
ch[root][1]=ch[oldroot][1];
clear(oldroot);
update(root);
}
else
{
int o=ch[root][1]>0;
root=ch[root][o];
clear(fa[root]);
fa[root]=0;
}
}
inline int find_order_of_key(int x)
{
int res=0,now=root;
while(1)
{
if(x<key[now])now=ch[now][0];
else
{
res+=size[ch[now][0]];
if(x==key[now]){splay(now,root);return res+1;}
res+=cnt[now];
now=ch[now][1];
}
}
}
inline int find_by_order(int x)
{
int now=root;
while(1)
{
if(x<=size[ch[now][0]])now=ch[now][0];
else
{
int temp=size[ch[now][0]]+cnt[now];
if(x<=temp)return key[now];
else{x-=temp;now=ch[now][1];}
}
}
}
void input()
{
int T=read<int>();
int opt,x;
while(T--)
{
opt=read<int>();x=read<int>();
if(opt==1)insert(x);
else if(opt==2)del(x);
else if(opt==3)printf("%d\n",find_order_of_key(x));
else if(opt==4)printf("%d\n",find_by_order(x));
else if(opt==5)
{
insert(x);
printf("%d\n",key[pre()]);
del(x);
}
else if(opt==6)
{
insert(x);
printf("%d\n",key[nex()]);
del(x);
}
}
}
int main()
{
file();
input();
return 0;
}
洛谷P3391 splay模板
#include<bits/stdc++.h>
using namespace std;
typedef int sign;
typedef long long ll;
#define For(i,a,b) for(register sign i=(sign)a;i<=(sign)b;++i)
#define Fordown(i,a,b) for(register sign i=(sign)a;i>=(sign)b;--i)
const int N=1e5+5;
bool cmax(sign &a,sign b){return (a<b)?a=b,1:0;}
bool cmin(sign &a,sign b){return (a>b)?a=b,1:0;}
template<typename T>T read()
{
T ans=0,f=1;
char ch=getchar();
while(!isdigit(ch)&&ch!=‘-‘)ch=getchar();
if(ch==‘-‘)f=-1,ch=getchar();
while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch-‘0‘),ch=getchar();
return ans*f;
}
void file()
{
#ifndef ONLINE_JUDGE
freopen("splay.in","r",stdin);
freopen("splay.out","w",stdout);
#endif
}
int ch[N][2],fa[N],size[N],rev[N],root,sz;
inline int get(int x){return x==ch[fa[x]][1];}
inline void update(int x){size[x]=1+size[ch[x][0]]+size[ch[x][1]];}
inline void rotate(int x,int &k)
{
int old=fa[x],oldfa=fa[old],o=get(x);
if(k==old)k=x;
else ch[oldfa][ch[oldfa][1]==old]=x;
fa[x]=oldfa;fa[old]=x;fa[ch[x][o^1]]=old;
ch[old][o]=ch[x][o^1];ch[x][o^1]=old;
update(x),update(old);
}
inline void splay(int x,int &k)
{
while(x!=k)
{
if(fa[x]!=k)rotate(get(x)^get(fa[x])?x:fa[x],k);
//printf("%d %d\n",x,k);
rotate(x,k);
}
}
#define mid ((l+r)>>1)
inline void build(int l,int r,int pre)
{
if(l>r)return;
ch[pre][mid>=pre]=mid;
fa[mid]=pre;size[mid]=1;
if(l==r)return;
build(l,mid-1,mid);build(mid+1,r,mid);
update(mid);
}
#undef mid
int n,m;
void input(){n=read<int>();m=read<int>();}
inline void rever(int x)
{
swap(ch[x][0],ch[x][1]);
rev[ch[x][0]]^=1;rev[ch[x][1]]^=1;
rev[x]=0;
}
int find(int x)
{
int now=root;
while(1)
{
if(rev[now])rever(now);
if(size[ch[now][0]]>=x)now=ch[now][0];
else
{
if(size[ch[now][0]]==x-1)return now;
x=x-size[ch[now][0]]-1;
now=ch[now][1];
}
}
}
void work()
{
int l,r;
root=(n+3)>>1;
build(1,n+2,root);
fa[root]=0;
while(m--)
{
l=read<int>();r=read<int>();
l=find(l);r=find(r+2);
splay(l,root);splay(r,ch[l][1]);
rev[ch[r][0]]^=1;
}
}
void out(int x)
{
if(rev[x])rever(x);
if(ch[x][0])out(ch[x][0]);
if(x>1&&x<n+2)printf("%d ",x-1);
if(ch[x][1])out(ch[x][1]);
}
int main()
{
file();
input();
work();
out(root);
return 0;
}
另外推荐一篇写得好的博客