【模板】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 ...
随机推荐
- solr linux配置
一.先在官网下载solr的最新版本或者你需求的版本1 目前我使用的是4.10.3(http://archive.apache.org/dist/lucene/solr/4.10.3/)2 复制到你的云 ...
- [转]微信小程序联盟 跳坑《一百八十一》设置API:wx.openSetting使用说明
本文转自:http://www.wxapp-union.com/forum.php?mod=viewthread&tid=4066 这个API解决了过去一个长久以来无法解决的问题,如何让用户重 ...
- jquery 使用整理
1. 如何创建嵌套的过滤器 //允许你减少集合中的匹配元素的过滤器, //只剩下那些与给定的选择器匹配的部分.在这种情况下, //查询删除了任何没(:not)有(:has) //包含class为“se ...
- WP手机短信导出方法和MSG格式文件阅读器的实现
最近想起来自己一直扔在抽屉里的Nokia920T里还存着珍贵的短信,觉得把它导出来存到电脑上比较稳妥也方便阅读.经过搜索找到一下方法:到应用市场里搜索contacts+message backup,安 ...
- Centos7 linux下 安装 Redis 5.0
网上找了很多文章,发现不全而且有些问题,安装很多次之后,总结一篇可以使用的,记录之. 环境:Centos7+Redis 5.0,如果环境不符合,本篇仅供参考. 1.准备工作 作者习惯软件安装包放在单独 ...
- Web开发:Bootstrap的应用
- [转]ubuntu 13.04 体验wine qq
1.首先安装最新版的wine1.52,没记错版本号应该是这个 sudo add-apt-repository ppa:ubuntu-wine/ppa sudo apt-get update sudo ...
- 设计模式入门,命令模式,c++代码实现
// test06.cpp : Defines the entry point for the console application.////设计模式第5章 命令模式#include "s ...
- java下double相乘精度丢失问题
比如 System.out.println(0.14*100); 输出: 14.000000000000002 解决方法: BigDecimal b = new BigDecimal(String.v ...
- IndexedDB(一:基本使用)
在HTML5本地存储--Web SQL Database提到过Web SQL Database实际上已经被废弃,而HTML5的支持的本地存储实际上变成了 Web Storage(Local Stora ...