平衡树学习笔记

已于 2022-07-12 14:14:56 修改
阅读量6.9k 收藏
点赞数
分类专栏： 算法 模板 文章标签： 1024程序员节 c++ 数据结构 算法
于 2022-03-06 16:14:43 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_49458360/article/details/123312131
版权

文章目录

平衡树

算法简介

平衡树是一种支持

  1. 插入一个整数 x。

  2. 删除一个整数x（若有多个相同的数，只删除一个）。

  3. 查询整数 x 的排名（排名定义为比当前数小的数的个数 +1）。

  4. 查询排名为 x 的数（如果不存在，则认为是排名小于 x 的最大数）。

  5. 求 x 的前驱（前驱定义为小于 x，且最大的数）。

  6. 求 x 的后继（后继定义为大于 x，且最小的数）。

等操作，并且支持在线的数据结构

分类

  1. 伪平衡树（并不是平衡树）
  2. Treap
  3. 替罪羊树
  4. AVL树
  5. 伸展树
  6. 红黑树
  7. 加权平衡树
  8. 2-3树
  9. AA树
  10. 节点大小平衡树

伪平衡树

注意：这并不是平衡树，只是可以支持大部分平衡树的操作的一种结构

实质：vector + lower_bound

vector

vector 是一种支持任意位置插入、删除和查询的一个结构。

定义方法std::vector<int>vec;//其中int可改为任意结构

它有以下几个常用函数：

  1. push_back(x)在尾部加入一个元素 x
  2. insert(pos,x) 在 pos 位置插入一个元素 x
  3. pop_back()在尾部删除一个元素
  4. erase(pos)在 pos 位置删除一个元素
  5. begin()返回第一个元素的位置
  6. end()返回最后一个元素的位置
  7. “operator[x]” 返回第 x 个元素
lower_bound

lower_bound 是 stl 自带的二分查找函数。

用法：std::lower_bound(/*起始地址*/,/*结束地址*/,/*所查询元素*/)

它会返回第一个小于等于所查询元素的位置。

实现
#include<bits/stdc++.h>
using namespace std;
int N,opt,x;
vector<int>vec;
int main(){
	scanf("%d",&N);
	while(N--){
		scanf("%d%d",&opt,&x);
		if(opt==1){vec.insert(lower_bound(vec.begin(),vec.end(),x),x);}
		if(opt==2){vec.erase(lower_bound(vec.begin(),vec.end(),x));}
		if(opt==3){printf("%d\n",lower_bound(vec.begin(),vec.end(),x)-vec.begin()+1);}
		if(opt==4){printf("%d\n",vec[x-1]);}
		if(opt==5){printf("%d\n",vec[lower_bound(vec.begin(),vec.end(),x)-vec.begin()-1]);}
		if(opt==6){printf("%d\n",*lower_bound(vec.begin(),vec.end(),x+1));}
	}
}
优缺点

优点：代码简短，不易打挂，易调试，支持在线。

缺点：由于 vector 和 lower_bound 的嵌套使用，时间复杂度较高。

适用于：

  1. 数据范围较小（一般 n ≤ 1 0 6 n\le10^6 n≤106 均可）

  2. 对常数要求不高的平衡树算法

Treap

Treap这个名字十分有内涵：

T r e e + H e a p = T r e + e a p = T r e a p \color{red}{Tree} \color{black}{+} \color{blue}{Heap} \color{black}{=} \color{red}{Tre} \color{black}{+} \color{blue}{eap} \color{black}{=} \color{red}{Tr}\color{purple}{e}\color{blue}{ap} Tree+Heap=Tre+eap=Treap

即 Treap = Tree(二叉搜索树) + Heap(堆)

这也是 Treap 能够平衡的关键。

基础操作
节点

对于每一个节点我们都会给它以下几个参数：

struct Treap{
    int l,//左儿子
    	r,//右儿子
    	val,//键值
    	ord,//随机值
    	siz,//子树大小
    	num;//同键值的数的个数
}t[Maxn];
旋转

旋转分为左旋（zig）和右旋（zag），如下图：

zig-zag

图1的中序遍历： D → B → E → A → C D\to B\to E\to A\to C D→B→E→A→C

图2的中序遍历： D → B → E → A → C D\to B\to E\to A\to C D→B→E→A→C

可以发现 zig 和 zag 是不改变树的中序遍历的

所以我们就可以在原树上放心的使用 zig 和 zag 来让树平衡

void zag(int &u){
    int tmp=t[u].l;
    t[u].l=t[tmp].r;
	t[tmp].r=u,u=tmp;
    pushup(t[u].r);pushup(u);
}

void zig(int &u){
    int tmp=t[u].r;
    t[u].r=t[tmp].l;
	t[tmp].l=u,u=tmp;
    pushup(t[u].l);pushup(u);
}
插入

插入操作与 BST 基本相同，对于一个已经拥有的节点，直接 num++ 即可。

对于一个新节点，我们先给他一个随机的 ord 值，按照 BST 的方法插入

回溯时判断是否进行旋转。

void insert(int &u,int x){
    if(!u){sz++,u=sz;t[u].siz=t[u].num=1;t[u].val=x,t[u].ord=rand();return;}
    if(t[u].val==x){t[u].num++;t[u].siz++;}
    else if(t[u].val<x){insert(t[u].r,x);if(t[t[u].r].ord<t[u].ord)zig(u);}
    else if(t[u].val>x){insert(t[u].l,x);if(t[t[u].l].ord<t[u].ord)zag(u);}
    pushup(u);
}
删除

删除和 BST 也是基本相同，对于 n u m > 1 num>1 num>1 num-- 。

否则将此节点移动到叶子，然后删除其与父节点的连边。

void del(int &u, int x){
    if(t[u].val==x){
        if(t[u].num>1){t[u].num--,t[u].siz--;}
        else if(!t[u].l){u=t[u].r;}
        else if(!t[u].r){u=t[u].l;}
        else if(t[t[u].l].ord<t[t[u].r].ord){zag(u);del(t[u].r,x);}
        else if(t[t[u].l].ord>t[t[u].r].ord){zig(u);del(t[u].l,x);}
    }
    else if(t[u].val>x){del(t[u].l,x);}
    else if(t[u].val<x){del(t[u].r,x);}
    pushup(u);
}
查询排名与元素

这两个都和 BST 完全一样。

查询元素的排名：

int queryrnk(int u, int x){
    if(!u)return 1;
    else if(t[u].val==x) return t[t[u].l].siz+1;
    else if(x>t[u].val) return t[t[u].l].siz+t[u].num+queryrnk(t[u].r,x);
    else if(x<t[u].val) return queryrnk(t[u].l,x);
}

查询排名的元素：

int querynum(int u, int x){
    if(x<=t[t[u].l].siz+t[u].num&&x>t[t[u].l].siz)return t[u].val;
    else if(x<=t[t[u].l].siz)return querynum(t[u].l,x);
    else return querynum(t[u].r,x-t[t[u].l].siz-t[u].num);
}
前驱和后继
int querypre(int u, int x){
    if(!u){return -(0x3fffffff);}
    if(t[u].val<x)return max(t[u].val,querypre(t[u].r,x));
    else return querypre(t[u].l,x);
}
int querysub(int u, int x){
    if(!u){return +(0x3fffffff);}
    if(t[u].val>x)return min(t[u].val,querysub(t[u].l,x));
    else return querysub(t[u].r,x);
}
实现
#include<bits/stdc++.h>
using namespace std;
#define Maxn 1000039
struct Treap{int l,r,val,ord,siz,num;}t[Maxn];
int N,op,x,sz=0,rt=1;
void pushup(int u){t[u].siz=t[t[u].l].siz+t[t[u].r].siz+t[u].num;}
void zig(int &u){
    int tmp=t[u].r;
    t[u].r=t[tmp].l;
	t[tmp].l=u,u=tmp;
    pushup(t[u].l);pushup(u);
}
void zag(int &u){
    int tmp=t[u].l;
    t[u].l=t[tmp].r;
	t[tmp].r=u,u=tmp;
    pushup(t[u].r);pushup(u);
}
void insert(int &u,int x){
    if(!u){sz++,u=sz;t[u].siz=t[u].num=1;t[u].val=x,t[u].ord=rand();return;}
    if(t[u].val==x){t[u].num++;t[u].siz++;}
    else if(t[u].val<x){insert(t[u].r,x);if(t[t[u].r].ord<t[u].ord)zig(u);}
    else if(t[u].val>x){insert(t[u].l,x);if(t[t[u].l].ord<t[u].ord)zag(u);}
    pushup(u);
}
void del(int &u, int x){
    if(t[u].val==x){
        if(t[u].num>1){t[u].num--,t[u].siz--;}
        else if(!t[u].l){u=t[u].r;}
        else if(!t[u].r){u=t[u].l;}
        else if(t[t[u].l].ord<t[t[u].r].ord){zag(u);del(t[u].r,x);}
        else if(t[t[u].l].ord>t[t[u].r].ord){zig(u);del(t[u].l,x);}
    }
    else if(t[u].val>x){del(t[u].l,x);}
    else if(t[u].val<x){del(t[u].r,x);}
    pushup(u);
}
int queryrnk(int u, int x){
    if(!u)return 1;
    else if(t[u].val==x) return t[t[u].l].siz+1;
    else if(x>t[u].val) return t[t[u].l].siz+t[u].num+queryrnk(t[u].r,x);
    else if(x<t[u].val) return queryrnk(t[u].l,x);
}
int querynum(int u, int x){
    if(x<=t[t[u].l].siz+t[u].num&&x>t[t[u].l].siz)return t[u].val;
    else if(x<=t[t[u].l].siz)return querynum(t[u].l,x);
    else return querynum(t[u].r,x-t[t[u].l].siz-t[u].num);
}
int querypre(int u, int x){
    if(!u){return -(0x3fffffff);}
    if(t[u].val<x)return max(t[u].val,querypre(t[u].r,x));
    else return querypre(t[u].l,x);
}
int querysub(int u, int x){
    if(!u){return +(0x3fffffff);}
    if(t[u].val>x)return min(t[u].val,querysub(t[u].l,x));
    else return querysub(t[u].r,x);
}
void init(){
	srand(time(0));
	sz++;t[sz].siz=t[sz].num=1;t[sz].val=-(0x3fffffff),t[sz].ord=rand();
	sz++;t[sz].siz=t[sz].num=1;t[sz].val=+(0x3fffffff),t[sz].ord=rand();
	t[1].r=2;pushup(1);
}
int main(){
	init();
    scanf("%d",&N);
    while(N--){
        scanf("%d%d",&op,&x);
        if(op==1){insert(rt,x);}
        if(op==2){del(rt,x);}
        if(op==3){printf("%d\n",queryrnk(rt,x)-1);}
        if(op==4){printf("%d\n",querynum(rt,x+1));}
        if(op==5){printf("%d\n",querypre(rt,x));}
        if(op==6){printf("%d\n",querysub(rt,x));}
    }
    return 0;
}
优缺点

优点：代码与 BST 差距不大，简单易调试，支持在线，速度快。

缺点：由于关键字是随机的，不一定是稳定的 log ⁡ n \log{n} logn

适用于：

  1. 无法使用 vector+lower_bound 的其他大部分的题
柴春阳
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
后缀平衡树学习笔记
Don't fake it till you made it
11-06 1411
给定一个空串SS 操作1 代表在SS前加入一个字母使之成为新SS 操作2 代表在询问在当前SS中有多少连续子串等于给定串TT假设我们已经有了串SS的后缀平衡树 插入一个字母cc 我们用SiS_i 代表原串SS从第ii个字符开始的后缀 则后缀 cScS与 任意一个后缀 SiS_i的大小关系可以用 c与SiS_i 的第一个字母的大小关系 以及 后缀S1S_1与后缀Si+1S_i+1
平衡树 - FHQ 学习笔记
EricQian的博客
03-31 2551
平衡树 - FHQ 学习笔记 主要参考万万没想到 的 FHQ-Treap学习笔记。 本片文章的姊妹篇：平衡树 - Splay 学习笔记。 感觉完全不会平衡树，又重新学习了一遍 FHQ，一口气把常见套路都学完了。 一、大致内容及分类 FHQ(???)，全称非旋转 Treap，是一种可以用于维护按权值、排名分裂的数据结构。它相比与 Splay 虽然常数较大，但是实现起来代码难度相对容易，而且由于它非...
[平衡树] 平衡树学习笔记
HeRaNO@HSAHRBNU
07-27 1053
好久不见！ 就差这个坑就补齐了！ 估计高三也不会太更博客了，不知道是不是NOIP前的最后一篇？ 在博客园的新博客欢迎关注！ 发现自己平衡树学了跟没学一样，退役了简单学了一发…… 平衡树有很多，比如AVL树，红黑树（RBTree）什么的，AVL被魔改成SBT？RBTree是系统中的平衡树。但是不常用不介绍了。 平衡树的本质是一棵二叉搜索树（Binary Search Tree，即BST），
平衡树学习笔记（2）——替罪羊树
kyrie lrving的博客
07-07 801
文章目录史上最暴力的平衡树——替罪羊树前言大致思路实现部分0.前置1.数组介绍2.内存池1.重构2.插入3.查询1.查询排名第x的数的值2.查询值为x的数的排名4.删除5.判断重构6.综合运用1.插入2.删除3.查询4.查找前驱5.查找后继一些废话总结==**（中考加油！）**== 史上最暴力的平衡树——替罪羊树 前言 ​ 想学替罪羊树很久了，刚开始接触平衡树的时候就久仰替罪羊树的大名，但是无奈经...
数据结构——学习笔记——入门必看【建议收藏】
热门推荐
刘鑫磊
02-09 41万+
（2）静态查找表&动态查找表的区别用法【顺序查找-线性查找-折半查找-二分查找-有序表查找-插值查找-斐波那契查找-索引顺序表查找-分块查找】单独使用顺序存储结构（即数组）无法很好地实现树的存储概念，不过如果充分利用顺序存储和链式存储结构的特点，则完全可以实现对数的存储结构的表示。深度为 2，则至多只有 3 个结点：根结点，根结点的左子树，根结点的右子树···指的是这样一个数列：1、1、2、3、5、8、13、21、……（3）四中常见的多路查找树【2-3 树$2-3-4 树$B 树$B+ 树】的区别用法。
【学习笔记】主席树
ModestCoder_的博客
05-11 1万+
1、前置知识 线段树、权值线段树、前缀和思想等 2、引入 主席树模板题 约定： 后面将第k小/大说成kthkthkth 解决什么问题： 给定一段区间，静态求区间kthkthkth 想想方法： 暴力：对于每一个询问，排个序，就行了，时间复杂度O(nmlogn)O(nmlogn)O(nmlogn) 莫队+树状数组：树状数组可以求给定区间kthkthkth，使用二分+树状数组，具体不展开，但是多个区间...
C++学习笔记:AVL树
m0_70088853的博客
03-08 986
C++学习笔记:AVL树
数据结构之二叉搜索树和二叉平衡树学习笔记
PrConstantin
10-28 1052
二叉搜索树（Binary Search Tree）具有下列性质的二叉树被称为二叉搜索树:（1）、若它的左子树不为空，则左子树上所有结点的值均小于它的根结点的值； （2）、若它的右子树不为空，则右子树上所有结点的值均大于它的根结点的值； （3）、它的左、右子树也分别为二叉查找树。 一些典型特点： - 而中序遍历二叉搜索树可得到一个关于key的有序序列，一个无序序列可以通过构造一棵二叉搜索树变成一个有
树与二叉树的学习笔记
qq_73106050的博客
04-16 1068
结点之间存在分支关系结点之间存在层次关系树的基本定义如下：树（Tree）是n（n≥0）个结点的有限集有且仅有一个特定的称为根（Root）的结点n = 0是一个空树当n＞1时，其余结点可分为m（m＞0）个互不相交的有限集T 1 、T 2 、……、T m ，其中每一个集合本身又是一棵树，并且称为根的子树。树如下图所示：注：根节点一定是唯一的，不可能存在多个根节点。子树的个数没有限制，但它们一定是互不相交的不可能出现下面的一个树。
codewhy学习笔记
05-22
3. 树结构：二叉树、平衡树（AVL树、红黑树）、B树、B+树等，它们在数据库索引、文件系统等方面有广泛应用。 4. 图算法：Dijkstra最短路径算法、Floyd-Warshall所有对最短路径算法、Prim最小生成树算法、Kruskal...
CCNP学习笔记
08-28
本CCNP学习笔记涵盖了多个核心网络概念和技术，包括OSPF（开放最短路径优先）、STP（生成树协议）、AAA（认证、授权和审计）、BGP（边界网关协议）、冗余技术、DHCP（动态主机配置协议）、ISIS（中间系统到中间系统...
adboost、gbdt、xgboost学习笔记
05-17
在机器学习领域中，决策树是一种常用的监督学习算法。它可以用来解决分类和回归问题。今天，我们将探讨三种基于决策树的boosting算法：Adaboost、GBDT和XGBoost。 Adaboost Adaboost是boosting家族中最早的一种...
监督学习算法学习笔记.pdf
06-03
决策树是监督学习中常见的分类算法，它以树状结构展示决策路径，每个内部节点表示一个属性测试，每个叶子节点代表一个类别。构建决策树的过程是通过不断分割数据，使分割后的子集尽可能纯净，即所有数据实例属于同一...
ACM学习笔记ACMACM
最新发布
05-08
2. **数据结构**：链表、树（二叉树、平衡树如AVL和红黑树）、图、堆、栈、队列、哈希表等数据结构的实现与应用。熟悉这些数据结构能帮助快速找到最优解法。 3. **数学知识**：离散数学、组合数学、数论、线性代数...
[set]考前复习-操作系统
柴春阳的博客
11-05 717
[set]考前复习-操作系统题面题目描述输入输出setset介绍性质使用模拟思路 题面 题目描述 写一个程序来模拟操作系统的进程调度。假设该系统只有一个CPUCPUCPU，每一个进程的到达时间，执行时间和运行优先级都是已知的。其中运行优先级用自然数表示，数字越大，则优先级越高。如果一个进程到达的时候CPUCPUCPU是空闲的，则它会一直占用CPUCPUCPU直到该进程结束。除非在这个过程中，有一个比它优先级高的进程要运行。在这种情况下，这个新的（优先级更高的）进程会占用CPUCPUCPU，而老的只有等待。如
[链表-贪心]中位数
柴春阳的博客
10-05 647
中位数题面题面描述输入输出算法分析暴力暴力思路核心代码链表链表思路链表+贪心链表+贪心思路核心代码 题面 题面描述 输入一个奇数nnn和长度为nnn的数组aaa，请你求出aaa数组前1,3,5……n1,3,5……n1,3,5……n个数的中位数 注：多组数据 输入 第一行 一个数ppp，表示有ppp组数据 下一行，两个数kkk和nnn，表示是第kkk组 接下来若干行，每行101010个数，共nnn个数 输出 第一行输出kkk和中位数的个数(((通常为n/2+1n/2+1n/2+1 ))) 接下输出中位数，10
[DP]音量调节
柴春阳的博客
11-02 534
[DP]音量调节题面题目描述输入输出样例解析暴力DPDP状态设计DP转移方程式代码 题面 题目描述 一个吉他手准备参加一场演出。他不喜欢在演出时始终使用同一个音量，所以他决定每一首歌之前他都需要改变一次音量。在演出开始之前，他已经做好一个列表，里面写着每首歌开始之前他想要改变的音量是多少。每一次改变音量，他可以选择调高也可以调低。 音量用一个整数描述。输入文件中整数beginLevelbeginLevelbeginLevel，代表吉他刚开始的音量，整数maxLevelmaxLevelmaxLevel，代表吉
[二分]第K大的数
柴春阳的博客
11-01 491
[二分]第K大的数题面题面描述输入输出样例解析暴力优化优化II代码 题面 题面描述 数组AAA和数组BBB，里面都有nnn个整数。 数组CCC共有n2n^2n2个整数，分别是： A0×B0,A0×B1......A0×Bn−1A_{0} \times B_{0},A_{0} \times B_{1} ...... A_{0} \times B_{n-1}A0​×B0​,A0​×B1​......A0​×Bn−1​ A1×B0,A1×B1......A1×Bn−1A_{1} \times B_{0},A_{1
决策树学习详解：从ID3到CART
"这篇文档是基于李航老师的《统计学习方法》中决策树章节的学习笔记，涵盖了决策树的基本概念、特征选择、决策树生成算法、剪枝策略以及CART算法等核心知识点。" 决策树是一种广泛应用的机器学习算法，它通过一系列...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值