平衡树模板
时间:2019-04-20
本文章向大家介绍平衡树模板,主要包括平衡树模板使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int N=1e6+10;
int ch[N][2]; //ch[][0] 表示左儿子 ch[][1] 表示右儿子
int f[N]; //节点的父亲节点
int sz[N]; //当前节点给所在的子树的节点个数
int cnt[N]; //当前节点所表示的值的个数
int root; //记录根节点的
int tot; //计算树中节点个数
int key[N]; //当前节点表示的值
void array_clear(int x)
{
f[x]=cnt[x]=ch[x][0]=ch[x][1]=key[x]=0;
}
bool get(int x)
{
return ch[f[x]][1]==x;
}
///0 左 1 右
void pushup(int x)
{
if(x)
{
sz[x]=cnt[x];
if(ch[x][0]) sz[x]+=sz[ch[x][0]];
if(ch[x][1]) sz[x]+=sz[ch[x][1]];
}
}
void rotates(int x)
{
int old=f[x],oldf=f[old],which=get(x);
ch[old][which]=ch[x][which^1];
f[ch[old][which]]=old;
///当前点的儿子过继给当前点的爸爸;
///同时处理父子两个方向上的信息
ch[x][which^1]=old;
f[old]=x;
///当前点变成曾经爸爸的爸爸;
f[x]=oldf;///爷爷变爸爸
if(oldf) ch[oldf][ch[oldf][1]==old]=x;///相等就是1,不等就是0
pushup(old);///维护信息
pushup(x);
}
void splay(int x)
{
for(int fa;fa=f[x];rotates(x))
if(f[fa])
rotates((get(x)==get(fa))?fa:x);
root=x;
}
void tree_insert(int x)
{
if(root==0)
{
tot++;
key[tot]=x;root=tot;
cnt[tot]=sz[tot]=1;
ch[tot][0]=ch[tot][1]=f[tot]=0;
return;
}
int now=root,fa=0;
while(true)
{
if(x==key[now])
{
cnt[now]++;
pushup(now);pushup(fa);splay(now);
return;
}
fa=now;now=ch[now][key[now]<x];
if(now==0)
{
tot++;
sz[tot]=cnt[tot]=1;
ch[tot][0]=ch[tot][1]=0;
ch[fa][x>key[fa]]=tot;
f[tot]=fa;
key[tot]=x;
pushup(fa);splay(tot);return;
}
}
}
int kth(int x)///找到排名为x的点
{
int now=root;
while(true)
{
if(ch[now][0]&&x<=sz[ch[now][0]])
now=ch[now][0];
else
{
int tmp=sz[ch[now][0]]+cnt[now];
if(x<=tmp)
return key[now];
x-=tmp;now=ch[now][1];
}
}
}
int rak(int x)///查询x的排名
{
int now=root,ans=0;
while(true)
{
if(x<key[now]) now=ch[now][0];
else
{
ans+=sz[ch[now][0]];
if(x==key[now])
{
splay(now);
return ans+1;
}
ans+=cnt[now];
now=ch[now][1];
}
}
}
int preve()
{
int now=ch[root][0];
while(ch[now][1]) now=ch[now][1];
return now;
}
int Nextt()
{
int now=ch[root][1];
while(ch[now][0]) now=ch[now][0];
return now;
}
void del(int x)
{
rak(x);
if(cnt[root]>1) {cnt[root]--; pushup(root); return ;}
if(!ch[root][0]&&!ch[root][1]) {array_clear(root);root=0;return; }
if(!ch[root][0])
{
int oldrt=root;
root=ch[root][1];
f[root]=0;
array_clear(oldrt);
return ;
}
else if(!ch[root][1])
{
int oldrt=root;
root=ch[root][0];
f[root]=0;
array_clear(oldrt);
return;
}
int oldrt=root;
int leftbig=preve();
splay(leftbig);
ch[root][1]=ch[oldrt][1];
f[ch[oldrt][1]]=root;
array_clear(oldrt);
pushup(root);
}
int main()
{
int n,op,a;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d %d",&op,&a);
if(op==1) tree_insert(a);
else if(op==2) del(a);
else if(op==3) printf("%d\n",rak(a));
else if(op==4) printf("%d\n",kth(a));
else if(op==5)
{
tree_insert(a);
printf("%d\n",key[preve()]);
del(a);
}
else
{
tree_insert(a);
printf("%d\n",key[Nextt()]);
del(a);
}
}
return 0;
}
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法
- java.lang.ClassNotFoundException:org.springframework.web.context.ContextLoaderListener问题解决
- ELK + Filebeat + Kafka 分布式日志管理平台搭建
- 一切皆是文件:UNIX,Linux 操作系統的設計哲學
- 【JS】687- 几行代码摸清楚上拉加载原理
- CTO 写的代码,全网被吐槽,真是绝了
- 定义一个函数,在该函数中可以实现任意两个整数的加法。java实现
- JDBC连接ORACLE的三种URL格式
- Centreon+Nagios实战第八篇——Nagios+Centreon添加监控服务
- Centreon+Nagios实战第九篇——利用nrpe插件监控本机
- 第四篇 CentOs7下安装Zabbix
- 第十三篇 zabbix创建Item
- 第十四篇 zabbix创建自定义Item
- 【性能】688- 前端性能优化——从 10 多秒到 1.05 秒
- 第十六篇 zabbix创建Trigger
- 如何通过程序(java代码)提高你的博客访问量