链接一下题目:luoguP3369[模板]普通平衡树(Treap/SBT)

平衡树解析

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<iomanip>
#include<algorithm>
#include<ctime>
#include<queue>
#include<stack>
#define lst long long
#define rg register
#define N 500050
#define Inf 2147483647
using namespace std; int use,root,tot;//操作数,根节点编号,树中元素总和
struct T{
int cnt;//这个数相等的数的数量
int size;//这棵子树上一共有几个元素
int fa,v;//父亲节点,当前点的权值
int ch[];//左(0)右(1)孩子
}ljl[N];//平衡树 inline int read()
{
rg int s=,m=;char ch=getchar();
while(ch!='-'&&(ch<''||ch>''))ch=getchar();
if(ch=='-')m=-,ch=getchar();
while(ch>=''&&ch<='')s=(s<<)+(s<<)+ch-'',ch=getchar();
return s*m;
} inline void Pushup(rg int now)//更新节点size的操作
{
ljl[now].size=ljl[ljl[now].ch[]].size+ljl[ljl[now].ch[]].size+ljl[now].cnt;
//当前节点的size是左孩子子树的size加上右孩子子树的size加上本节点相等的数的数量
} inline void rotate(rg int x)//把x往上转
{//定义:x的相对位置为x属于y的?孩子
rg int y=ljl[x].fa;//父亲
rg int z=ljl[y].fa;//祖父
rg int k=ljl[y].ch[]==x;//x的相对位置
ljl[z].ch[ljl[z].ch[]==y]=x;//把x转到y的位置上去
ljl[x].fa=z;//x的爸爸变成了z
ljl[y].ch[k]=ljl[x].ch[k^];//y的x的相对位置的那个孩子变成x的x的相对位置的另一个孩子
ljl[ljl[x].ch[k^]].fa=y;//……的爸爸变成y
ljl[x].ch[k^]=y;//x的相对位置的另一个孩子变成y
ljl[y].fa=x;//y的爸爸变成x
Pushup(x),Pushup(y);//更新一下节点数量
} inline void splay(rg int x,rg int goal)//把x转到goal下面,如果goal=0,那么就是转到根节点
{
while(ljl[x].fa!=goal)//如果x的父亲不是goal,目标没有达成,就要继续转
{
rg int y=ljl[x].fa;//父亲
rg int z=ljl[y].fa;//祖父
if(z!=goal)//如果z存在的话
{
(x==ljl[y].ch[])^(y==ljl[z].ch[])?rotate(x):rotate(y);
//如果x和y分别是y和z的同一孩子,就把y往上转
//如果x和y分别是y和z的不同孩子,就把x往上转
}
rotate(x);//最后一定会要把x在网上转一次
}
if(!goal)root=x;//更新根节点
} void Insert(rg int x)//插入x
{
rg int now=root,fa=;//从根开始找,根的父亲是0
while(ljl[now].v!=x&&now)//只要还没有找到这个数字,且当前这个位置有数,就继续找
{
fa=now;//爸爸变成现在的节点
now=ljl[now].ch[x>ljl[now].v];//如果x比now大,就找now的右孩子,小则左孩子
}
if(now)ljl[now].cnt++;//如果存值的位置存在,就直接在计数器上加1
else//否则
{
now=++tot;//增加一个新位置
if(fa)ljl[fa].ch[x>ljl[fa].v]=now;//如果父亲存在(我不是根),那我的父亲的儿子是我
ljl[now].v=x;//权值
ljl[now].fa=fa;//父亲
ljl[now].cnt=;//计数器
ljl[now].size=;//子树大小
ljl[now].ch[]=ljl[now].ch[]=;//没有孩子
}
splay(now,);//把当前位置转到根节点,以维持树的平衡
} inline void find(rg int x)//找x的位置,把它转到根节点,方便之后的计算
{
rg int now=root;//从根开始找
if(!root)return;//如果是空树,还找个屁
while(x!=ljl[now].v&&ljl[now].ch[x>ljl[now].v])//如果还没有找到那个点,且我还有符合的儿子
now=ljl[now].ch[x>ljl[now].v];//就跳转到我的儿子继续找
splay(now,);//转到树根去
} inline int Next(rg int x,rg int f)//找x的前驱(0)后继(1)
{
find(x);//先找到x的位置,可能树顶不是x,是和x值接近的那个元素
int now=root;//从根开始找
if(ljl[now].v>x&&f)return now;//如果大于x且我们要找后继,那就是他了
if(ljl[now].v<x&&!f)return now;//如果小于x且我们要找前驱,那就是他了
now=ljl[now].ch[f];//那我们从符合条件的儿子开始跳
while(ljl[now].ch[f^])now=ljl[now].ch[f^];//不断的往最优的方向跳
return now;//返回位置
} inline void Delete(rg int x)//删掉x
{
rg int qq=Next(x,);//找到前驱
rg int hj=Next(x,);//找到后继
splay(qq,),splay(hj,qq);//把前驱转到树根,把后继转到前驱下面
int del=ljl[hj].ch[];//那么x就是后继的左儿子
if(ljl[del].cnt>)//如果x的计数器大于1
{
ljl[del].cnt--;//让x的计数器--
splay(del,);//转到树根保持平衡
}
else ljl[hj].ch[]=;//直接删除x
} inline int kth(rg int x)//找第x小的数
{
rg int now=root;//从根开始找
if(ljl[now].size<x)return ;//如果排名都超过总数了…………
while()//嘿嘿,一直找
{
rg int ls=ljl[now].ch[];//左孩子
if(ljl[ls].size+ljl[now].cnt<x)//如果排名比左孩子总元素数和与我相等的数总和还大
{
x-=ljl[ls].size+ljl[now].cnt;//就减去前面的元素数
now=ljl[now].ch[];//去右孩子上找这个排名
}
else
if(ljl[ls].size>=x)now=ls;//如果左孩子里包括它
else return ljl[now].v;//那就在这个点上了,返回
}
} int main()
{
use=read();
Insert(Inf),Insert(-Inf);
for(rg int i=;i<=use;++i)
{
rg int opt=read(),x=read();
if(opt==)Insert(x);
if(opt==)Delete(x);
if(opt==)find(x),printf("%d\n",ljl[ljl[root].ch[]].size);
if(opt==)printf("%d\n",kth(x+));
if(opt==)printf("%d\n",ljl[Next(x,)].v);
if(opt==)printf("%d\n",ljl[Next(x,)].v);
}
return ;
}

最新文章

  1. JAVA程序员常用软件整理下载
  2. Oracle学习笔记五 SQL命令(三):Group by、排序、连接查询、子查询、分页
  3. 《简明python教程》笔记一
  4. javascript之Dorm
  5. Win8环境WPF打开和关闭软键盘
  6. KnockOut.js入门示例详解
  7. P and V
  8. HDU5829 NTT
  9. GUI编程笔记(java)04:GUI(HelloWorld)窗体案例
  10. HDU 2102 A计划(DFS)
  11. python 标准库 -- subprocess
  12. Openfire的web插件开发
  13. Java编程学习知识点分享 入门必看
  14. Ntrip协议简介(转)
  15. 单工程搭建springmvc+spring+mybatis(maven,idea)
  16. SQL将完整时间字段截取到年月日
  17. 6. Uniforms in American&#39;s Eyes 美国人眼里的制服
  18. bzoj千题计划232:bzoj4727: [POI2017]Turysta
  19. 制作高仿QQ的聊天系统(上)—— 布局文件 &amp; 减少过度绘制
  20. 解决IE6下浮层遮盖select刺穿的问题

热门文章

  1. kali优化配置(2)
  2. Windows 好用的护眼软件
  3. Linux性能优化从入门到实战:03 CPU篇:CPU上下文切换
  4. JFreeChart使用
  5. CF527E Data Center Drama
  6. SSM三大框架整合梳理
  7. 对getBoundingClientRect属性的研究
  8. ConversionService HttpMessageConverter
  9. leetcode_1292. Maximum Side Length of a Square with Sum Less than or Equal to Threshold_[二维前缀和]
  10. ward&#39;s method分层聚类凝聚法