【模板】Splay总结
rentenglong大佬写的splay的梳理使我受益颇丰,特此做出一定的总结。
数据结构
定义了一个struct结构体。
为了在splay操作下储存根节点,我们宏定义了root 为 tree[0].ch[1],则我们把tree[0]看作了整棵树的超级根,而tree[0]的右儿子为排序树的根节点,那么一定有tree[0].v < tree[tree[0].ch[1]].v,所以我们需要将tree[0].v赋值为无穷小。
struct Node{
int v, father; //v 保存的值, fa 该点的父亲结点
int ch[]; //ch[0] 左孩子(小于根) ch[1] 右孩子(大于根)
int sum; //以该结点为子树的所有数的出现次数
int recy; //该结点所代表的数的出现次数(记录重复次数)
}tree[MAX];
另外我们用两个变量n(搜索树的结点个数),points(整棵树中所有数的储存次数)。
更新当前结点的sum值
在进行splay操作后整棵树的结构可能会发生改变,那么有些结点的sum值就有可能会发生改变,我们需要维护sum的原有定义,所以我们要把当前结点的sum给更新一下。
void updata(int x)
{
tree[x].sum = tree[tree[x].ch[]].sum + tree[tree[x].ch[]].sum + tree[x].recy;
}
获取父子关系
用于确定当前结点是父亲结点的左孩子(ch[0])还是右孩子(ch[1])。左孩子小于父亲结点,右孩子大于父亲结点。
int identify(int x)
{
return tree[tree[x].father].ch[] == x ? : ;
}
建立父子关系
用于连接两个点,其中一个点是另一个点的孩子。connect(x, f, son(0/1))表示 x 是 f 的 son(0/1) 孩子(0表示左,1表示右)。
void connect(int x, int f, int son)
{
tree[x].father = f;
tree[f].ch[son] = x;
}
旋转操作
所谓旋转指的是把指定结点向上移动一级然后将其父亲结点作为他的儿子。
通过异或操作可以进行。
void rotate(int x)
{
int y = tree[x].father;
int mroot = tree[y].father;
int mrootson = identify(y);
int yson = identify(x);
int b = tree[x].ch[yson ^ ];
connect(b, y, yson), connect(y, x, (yson ^ )), connect(x, mroot, mrootson);
updata(y), updata(x);
}
伸展操作
把我们要用的结点旋转到根上来。
考虑双旋和单旋,节省时间。
void splay(int at, int to)
{
to = tree[to].father;
for(; tree[at].father != to;)
{
int up = tree[at].father;
if(tree[up].father == to) rotate(at);
else if(identify(up) == identify(at)) rotate(up), rotate(at);
else rotate(at), rotate(at);
}
}
新建结点&破坏处理废物结点
int crepoint(int v, int father)
{
n ++;
tree[n].v = v;
tree[n].father = father;
tree[n].sum = tree[n].recy = ;
return n;
}
void destory(int x)
{
tree[x].v = tree[x].ch[] = tree[x].ch[] = tree[x].father = tree[x].recy = tree[x].sum = ;
if(x == n) n --;
}
内置的插入结点
int build(int v)
{
points ++;
if(n == )
{
root = ;
crepoint(v, );
}
else
{
int now = root;
for(;;)
{
if(now) tree[now].sum ++;
if(v == tree[now].v)
{
tree[now].recy ++;
return now;
}
int Next = v < tree[now].v ? : ;
if(!tree[now].ch[Next])
{
crepoint(v, now);
tree[now].ch[Next] = n;
return n;
}
now = tree[now].ch[Next];
}
}
return ;
}
外部调用的查找结点
int find(int v)
{
int now = root;
for(;;)
{
if(tree[now].v == v)
{
splay(now, root);
return now;
}
int Next = v < tree[now].v ? : ;
if(!tree[now].ch[Next]) return ;
now = tree[now].ch[Next];
}
}
外部调用的插入结点&删除结点
void Push(int v)
{
int add = build(v);
splay(add, root);
}
void Pop(int v)
{
int deal = find(v);
if(!deal) return;
points --;
if(tree[deal].recy > )
{
tree[deal].recy --;
tree[deal].sum --;
return;
}
if(!tree[deal].ch[])
{
root = tree[deal].ch[];
tree[root].father = ;
}
else
{
int Left = tree[deal].ch[];
for(; tree[Left].ch[];) Left = tree[Left].ch[];
splay(Left, tree[deal].ch[]);
int Right = tree[deal].ch[];
connect(Right, Left, ), connect(Left, , );
updata(Left);
}
destory(deal);
}
查询第v大&查询v是第几大
int Rank(int v)//查询v的排名
{
int ans = , now = root;
for(;;)
{
if(tree[now].v == v) return ans + tree[tree[now].ch[]].sum + ;
if(now == ) return ;
if(v < tree[now].v) now = tree[now].ch[];
else{
ans = ans + tree[tree[now].ch[]].sum + tree[now].recy;
now = tree[now].ch[];
}
}
if(now) splay(now, root);
return ;
}
int At_rank(int x)//查询排名为x的数
{
if(x > points) return -INF;
int now = root;
for(;;)
{
int minused = tree[now].sum - tree[tree[now].ch[]].sum;
if(x > tree[tree[now].ch[]].sum && x <= minused) break;
if(x < minused) now = tree[now].ch[];
else{
x = x - minused;
now = tree[now].ch[];
}
}
splay(now, root);
return tree[now].v;
}
查找前驱&后继
int Upper(int v)
{
int now = root;
int result = INF;
for(; now;)
{
if(tree[now].v > v && tree[now].v < result) result = tree[now].v;
if(v < tree[now].v) now = tree[now].ch[];
else now = tree[now].ch[];
}
return result;
}
int Lower(int v)
{
int now = root;
int result = -INF;
for(; now;)
{
if(tree[now].v < v && tree[now].v > result) result = tree[now].v;
if(v > tree[now].v) now = tree[now].ch[];
else now = tree[now].ch[];
}
return result;
}
整段代码(洛谷P3369)
#include<bits/stdc++.h>
using namespace std; const int MAX = ;
const int INF = ; class Splay
{
#define root tree[0].ch[1]
private:
struct Node{
int v, father; //v 保存的值, fa 该点的父亲结点
int ch[]; //ch[0] 左孩子(小于根) ch[1] 右孩子(大于根)
int sum; //以该结点为子树的所有数的出现次数
int recy; //该结点所代表的数的出现次数(记录重复次数)
}tree[MAX];
int n = , points = ;
void updata(int x)
{
tree[x].sum = tree[tree[x].ch[]].sum + tree[tree[x].ch[]].sum + tree[x].recy;
}
int identify(int x)
{
return tree[tree[x].father].ch[] == x ? : ;
}
void connect(int x, int f, int son)
{
tree[x].father = f;
tree[f].ch[son] = x;
}
void rotate(int x)
{
int y = tree[x].father;
int mroot = tree[y].father;
int mrootson = identify(y);
int yson = identify(x);
int b = tree[x].ch[yson ^ ];
connect(b, y, yson), connect(y, x, (yson ^ )), connect(x, mroot, mrootson);
updata(y), updata(x);
}
void splay(int at, int to)
{
to = tree[to].father;
for(; tree[at].father != to;)
{
int up = tree[at].father;
if(tree[up].father == to) rotate(at);
else if(identify(up) == identify(at)) rotate(up), rotate(at);
else rotate(at), rotate(at);
}
}
int crepoint(int v, int father)
{
n ++;
tree[n].v = v;
tree[n].father = father;
tree[n].sum = tree[n].recy = ;
return n;
}
void destory(int x)
{
tree[x].v = tree[x].ch[] = tree[x].ch[] = tree[x].father = tree[x].recy = tree[x].sum = ;
if(x == n) n --;
}
int build(int v)
{
points ++;
if(n == )
{
root = ;
crepoint(v, );
}
else
{
int now = root;
for(;;)
{
if(now) tree[now].sum ++;
if(v == tree[now].v)
{
tree[now].recy ++;
return now;
}
int Next = v < tree[now].v ? : ;
if(!tree[now].ch[Next])
{
crepoint(v, now);
tree[now].ch[Next] = n;
return n;
}
now = tree[now].ch[Next];
}
}
return ;
}
public :
int getroot() {return root;}
void init()
{
tree[].v = -INF;
}
int find(int v)
{
int now = root;
for(;;)
{
if(tree[now].v == v)
{
splay(now, root);
return now;
}
int Next = v < tree[now].v ? : ;
if(!tree[now].ch[Next]) return ;
now = tree[now].ch[Next];
}
}
void Push(int v)
{
int add = build(v);
splay(add, root);
}
void Pop(int v)
{
int deal = find(v);
if(!deal) return;
points --;
if(tree[deal].recy > )
{
tree[deal].recy --;
tree[deal].sum --;
return;
}
if(!tree[deal].ch[])
{
root = tree[deal].ch[];
tree[root].father = ;
}
else
{
int Left = tree[deal].ch[];
for(; tree[Left].ch[];) Left = tree[Left].ch[];
splay(Left, tree[deal].ch[]);
int Right = tree[deal].ch[];
connect(Right, Left, ), connect(Left, , );
updata(Left);
}
destory(deal);
}
int Rank(int v)
{
int ans = , now = root;
for(;;)
{
if(tree[now].v == v) return ans + tree[tree[now].ch[]].sum + ;
if(now == ) return ;
if(v < tree[now].v) now = tree[now].ch[];
else{
ans = ans + tree[tree[now].ch[]].sum + tree[now].recy;
now = tree[now].ch[];
}
}
if(now) splay(now, root);
return ;
}
int At_rank(int x)
{
if(x > points) return -INF;
int now = root;
for(;;)
{
int minused = tree[now].sum - tree[tree[now].ch[]].sum;
if(x > tree[tree[now].ch[]].sum && x <= minused) break;
if(x < minused) now = tree[now].ch[];
else{
x = x - minused;
now = tree[now].ch[];
}
}
splay(now, root);
return tree[now].v;
}
int Upper(int v)
{
int now = root;
int result = INF;
for(; now;)
{
if(tree[now].v > v && tree[now].v < result) result = tree[now].v;
if(v < tree[now].v) now = tree[now].ch[];
else now = tree[now].ch[];
}
return result;
}
int Lower(int v)
{
int now = root;
int result = -INF;
for(; now;)
{
if(tree[now].v < v && tree[now].v > result) result = tree[now].v;
if(v > tree[now].v) now = tree[now].ch[];
else now = tree[now].ch[];
}
return result;
}
#undef root
};
Splay F; inline int read()
{
int x = , w = ; char ch= getchar();
for(; !isdigit(ch); w |= (ch == '-'), ch = getchar());
for(; isdigit(ch); x = (x << ) + (x << ) + (ch ^ ), ch = getchar());
return w ? -x : x;
} int main()
{
int opt, n, x;
//scanf("%d", &n);
n = read();
F.init();
for(int i = ; i <= n; ++ i)
{
//scanf("%d%d", &opt, &x);
opt = read(), x = read();
switch(opt)
{
case : F.Push(x); break;
case : F.Pop(x); break;
case : printf("%d\n", F.Rank(x)); break;
case : printf("%d\n", F.At_rank(x)); break;
case : printf("%d\n", F.Lower(x)); break;
case : printf("%d\n", F.Upper((x))); break;
}
}
return ;
}
【模板】Splay总结的更多相关文章
- 算法模板——splay区间反转 2
实现功能:同splay区间反转 1(基于BZOJ3223 文艺平衡树) 这次改用了一个全新的模板(HansBug:琢磨了我大半天啊有木有),大大简化了程序,同时对于splay的功能也有所完善 这里面没 ...
- 算法模板——splay区间反转 1
实现的功能:将序列区间反转,并维护 详见BZOJ3223 var i,j,k,l,m,n,head,a1,a2:longint; s1:ansistring; a,b,c,d,fat,lef,rig: ...
- 【luogu P3369 普通平衡树(Treap/SBT)】 模板 Splay
题目链接:https://www.luogu.org/problemnew/show/P3369 #include <cstdio> #include <algorithm> ...
- 洛谷 P3391 模板Splay
#include<bits/stdc++.h> using namespace std; #define maxn 200000 int read() { ,w=; ;ch=getchar ...
- [模板] Splay
欠了好久的Splay,以后就它了. 默写真不容易,过几天估计就忘了.. 整个Splay真的精妙,不拖泥带水那种.. 前驱后继之所以不能用rk转到根,是因为这个数不一定存在.. kth中<=老忘记 ...
- 模板—splay
#include<iostream> #include<cstdio> #define cin(x) scanf("%d",&x) using na ...
- 模板——Splay
$Splay$ #include <bits/stdc++.h> #define inf (int)1e9 using namespace std; const int N=1e5+100 ...
- Splay 伸展树
废话不说,有篇论文可供参考:杨思雨:<伸展树的基本操作与应用> Splay的好处可以快速分裂和合并. ===============================14.07.26更新== ...
- [NOI2003][bzoj1507] 文本编辑器 editor [splay]
其实看明白了就是一道水题 毕竟模板 splay敲一发,插入一个串的时候先把它构建成一棵平衡树,再挂到原来的splay上面去即可 没别的了,上代码 #include<iostream> #i ...
- splay最终模板
来自wjmzbmr的splay模板 #include<cstdio> #include<iostream> #include<algorithm> using na ...
随机推荐
- easyui导出当前datagrid数据(含表头)
JS代码 //导出当前DataGrid数据 function doExportCommon() { var list = getCheckedRowCommon(); var exportList = ...
- jquery获取子元素
Jquery获取子元素的方法有2种,分别是children()方法和find()方法. 下面我们分别来使用这两种方法,看看它们有何差异. children()方法:获取该元素下的直接子集元素 find ...
- CSS实现响应式布局(自动拆分几列)
1.css代码 <style type="text/css"> .container{ margin-top: 10px; } .outerDiv{ float:lef ...
- HTML语言中img标签的alt属性和title属性的作用与区别
alt属性是在你的图片因为某种原因不能加载时在页面显示的提示信息,它会直接输出在原本加载图片的地方,而title属性是在你鼠标悬停在该图片上时显示一个小提示,鼠标离开就没有了,有点类似jQuery的h ...
- 第7章 征服CSS3选择器(下)
:enabled选择器 在Web的表单中,有些表单元素有可用(":enabled")和不可用(":disabled")状态,比如输入框,密码框,复选框等.在默认 ...
- 将ojdbc 添加到maven
去oracle官网下载jar包 然后在jar包所在目录输入maven命令 mvn install:install-file -DgroupId=com.oracle -DartifactId=ojdb ...
- BZOJ3992: [SDOI2015]序列统计(NTT 原根 生成函数)
题意 题目链接 给出大小为\(S\)的集合,从中选出\(N\)个数,满足他们的乘积\(\% M = X\)的方案数 Sol 神仙题Orz 首先不难列出最裸的dp方程,设\(f[i][j]\)表示选了\ ...
- ArcSDE10.2.2使用SQL操作ST_Geometry时报ORA-28579或ORA-20006错误
ArcSDE10.2.2使用SQL操作ST_Geometry时报ORA-28579或ORA-20006错误 1.测试环境说明 ArcSDE版本:10.2.2 Oracle版本:12.1.0.1和11. ...
- Python爬虫教程-14-爬虫使用filecookiejar保存cookie文件(人人网)
Python爬虫教程-14-爬虫使用filecookiejar保存cookie文件(人人网) 上一篇介绍了利用CookieJar访问人人网,本篇将使用filecookiejar将cookie以文件形式 ...
- EM(期望最大化)算法初步认识
不多说,直接上干货! 机器学习十大算法之一:EM算法(即期望最大化算法).能评得上十大之一,让人听起来觉得挺NB的.什么是NB啊,我们一般说某个人很NB,是因为他能解决一些别人解决不了的问题.神为什么 ...