手撕AVL树

news2025/7/20 12:54:25

目录

一、概念

二、 结点的定义

2.1 键值对pair

2.2 定义细节

三、 AVL树的插入操作

3.1 平衡因子调整规则

3.2 旋转规则

3.2.1 新节点插入较高左子树的左侧 — 左左:右单旋 

3.2.2 新节点插入较高右子树的右侧 — 右右:左单旋

3.2.3 新节点插入较高左子树的右侧 — 左右:左右双旋

3.2.4 新节点插入较高右子树的左侧 — 右左:右左双旋

四、 AVL树的删除操作

五、 AVL树性能分析

六、 完整代码


一、概念

二叉搜索树虽可以缩短查找的效率,但若数据有序或接近有序二叉搜索树将退化为单支树,查找元素相当于在链表中搜索元素,效率低下。

因此,两位俄罗斯的数学家G.M.Adelson-Velskii和E.M.Landis在1962年发明了一种解决上述问题的方法:当向二叉搜索树中插入新结点后,若能保证每个结点的左右子树高度之差的绝对值不超过1(需要对树中的结点进行调整),即可降低树的高度,从而减少平均搜索长度。
 

一棵AVL树是空树 或 具有以下性质的二叉搜索树:
1. 它的左右子树都是AVL树
2. 左右子树高度之差(简称平衡因子)的绝对值不超过1(-1/0/1)    右子树高度减去左子树高度
3. 若一棵二叉搜索树是高度平衡的,其就是AVL树。
    若它有n个结点,其高度可保持在log_2 n,搜索时间复杂度O(log_2 n)
4. 不允许键值冗余

二、 结点的定义

2.1 键值对pair

pair是用来表示具有一一对应关系的一种结构,该结构中一般只包含两个成员变量key和value,key代表键值,value表示与key对应的信息

SGI-STL中对于pair的定义:

template <class T1, class T2>
struct pair
{
    typedef T1 first_type;
    typedef T2 second_type;
    T1 first;
    T2 second;
    pair(): first(T1()), second(T2()) {}
    pair(const T1& a, const T2& b): first(a), second(b) {}
};

使用时需要显示指定元素类型会导致代码过长,并且频繁使用时较为麻烦,于是出现make_pair进行自动类型推导。其定义为:

template <class T1,class T2>
pair<T1,T2> make_pair (T1 x, T2 y)
{    
    return ( pair<T1,T2>(x,y) );
}

2.2 定义细节

结点中存储了左子树指针、右子树指针和父指针以及一个键值对(即数据域)、平衡因子。不过平衡因子并不是必要的,没有平衡因子同样可以实现AVL树。

struct AVLTreeNode {
	AVLTreeNode(const pair<K,V>& kv) :
            _parent(nullptr), _left(nullptr), 
            _right(nullptr),_data(kv),_balance_factor(0) {}

	AVLTreeNode<K, V>* _parent;
	AVLTreeNode<K, V>* _left;
	AVLTreeNode<K, V>* _right;

	pair<K, V> _data;
	int _balance_factor;//平衡因子
};

普通二叉树和搜索二叉树都是二叉链(没有父指针),为什么AVL树会需要使用三叉链呢?

这个问题可以从后面的讲解得到答案。
插入和删除结点、旋转 、平衡因子调整,这些操作都会需要频繁的使用父指针,使用三叉链可以减少查找父结点的时间复杂度并且使得AVL树的实现更加方便。

三、 AVL树的插入操作

AVL本质上就是具有特殊性质的二叉搜索树,其插入操作与二叉搜索树较为相似,不过更为复杂。

AVL树的插入过程可以分为两步
1. 按照二叉搜索树的方式插入新节点    2. 调整节点的平衡因子

bool insert(const pair<K, V>& kv) {
    //第一步: 按照二叉搜索树的方式插入新节点 
	if (_root == nullptr) {
		_root = new TreeNode(kv);
		return true;
	}

	TreeNode* parent = nullptr;
	TreeNode* cur = _root;
	while (cur != nullptr) {
		if (kv.first > cur->_data.first) {
			parent = cur;
			cur = cur->_right;
		}
		else if (kv.first < cur->_data.first) {
			parent = cur;
			cur = cur->_left;
		}
		else return false;
	}
	cur = new TreeNode(kv);
	if (kv.first > parent->_data.first) {
		parent->_right = cur;
    }
	else { //kv.first < parent->_data.first)
		parent->_left = cur;
	}
	cur->_parent = parent;
    
    //调整平衡因子
    //………………    
}

但是平衡因子是如何调整的呢?

3.1 平衡因子调整规则

1. 若新增结点在其父结点的左边,则父结点的平衡因子 -1;若新增结点在其父结点的右边,则父结点的平衡因子 +1。

2. 更新后,若父结点的平衡因子为1 或 -1,说明插入前父结点的平衡因子为0,插入后父结点所在子树的高度发生变化,需要继续向上更新

3.更新后,若父结点的平衡因子为0,说明插入前父结点的平衡因子为1 或 -1,插入到了父结点矮的一边,父结点所在子树的高度并未发生变化,所以不需继续向上更新

4.更新后,若父结点的平衡因子为2 或 -2,说明插入前父结点的平衡因子为1 或 -1(达到平衡临界值),插入后已经破坏平衡,此时父结点所子树需要进行旋转处理

5.更新后,若父结点的平衡因子的绝对值大于2(理论上而言不可能),说明插入前该树就不是AVL树,需检查之前的操作。

while (parent != nullptr){
    //规则1
    if (cur == parent->_right) ++parent->_balance_factor;
	else --parent->_balance_factor;
    
    //规则2
	if(parent->_balance_factor == 0) break;
    
    //规则3
	else if (abs(parent->_balance_factor) == 1) {
		cur = parent;
		parent = parent->_parent;
	}

    //规则4
	else if (abs(parent->_balance_factor) == 2) {
		//需要旋转
        //………………
	}

    //规则5
    else {
		assert(false);
	}
    return true;
}

3.2 旋转规则

AVL树的旋转可以分为四种:

3.2.1 新节点插入较高左子树的左侧 — 左左:右单旋 

//发生右旋的判断条件
if (parent->_balance_factor == -2 && cur->_balance_factor == -1) {
	rotate_right(parent);
}
void rotate_right(TreeNode* parent) {
	TreeNode* subL = parent->_left;
	TreeNode* subLR = subL->_right;
	TreeNode* pparent = parent->_parent;//发生旋转的子树的父结点

	parent->_left = subLR;
	if (subLR != nullptr) subLR->_parent = parent;//h可能为0,即空树
	subL->_right = parent;
	parent->_parent = subL;

    //父结点所在的子树发生右旋转后,该子树的根结点发生改变
	if (_root == parent) {
		_root = subL;
		subL->_parent = nullptr;
	}
	else {
		if (pparent->_left == parent) pparent->_left = subL;
		else pparent->_right = subL;
		subL->_parent = pparent;
	}

    //更新平衡因子
	subL->_balance_factor = parent->_balance_factor = 0;
}

3.2.2 新节点插入较高右子树的右侧 — 右右:左单旋

//发生左旋的判断条件
if (parent->_balance_factor == 2 && cur->_balance_factor == 1) {
    rotate_left(parent);
}
void rotate_right(TreeNode* parent) {
	TreeNode* subL = parent->_left;
	TreeNode* subLR = subL->_right;
	TreeNode* pparent = parent->_parent;//发生旋转的子树的父结点

	parent->_left = subLR;
	if (subLR != nullptr) subLR->_parent = parent;//h可能为0,即空树
	subL->_right = parent;
	parent->_parent = subL;

    //parent所在的子树发生左旋转后,该子树的根结点发生变化
	if (_root == parent) {
		_root = subL;
		subL->_parent = nullptr;
	}
	else {
		if (pparent->_left == parent) pparent->_left = subL;
		else pparent->_right = subL;
		subL->_parent = pparent;
	}
    
    //更新平衡因子
	subL->_balance_factor = parent->_balance_factor = 0;
}

3.2.3 新节点插入较高左子树的右侧 — 左右:左右双旋

void rotate_left_right(TreeNode* parent) {
	TreeNode* subL = parent->_left;
	TreeNode* subLR = subL->_right;
	int bf = subLR->_balance_factor;

	rotate_left(parent->_left);
	rotate_right(parent);

	//更新平衡因子
	subLR->_balance_factor = 0;

    //情况一
	if (bf == 1) {
		parent->_balance_factor = 0;
		subL->_balance_factor = -1;
	}
    //情况二
	else if (bf == -1) {
		parent->_balance_factor = 1;
		subL->_balance_factor = 0;
	}
    //情况三
	else if (bf == 0) {
		parent->_balance_factor = 0;
		subL->_balance_factor = 0;
	}
	else assert(false);
}

3.2.4 新节点插入较高右子树的左侧 — 右左:右左双旋

void rotate_right_left(TreeNode* parent) {
	TreeNode* subR = parent->_right;
	TreeNode* subRL = subR->_left;
	int bf = subRL->_balance_factor;

	rotate_right(parent->_right);
	rotate_left(parent);

    //调整平衡因子
	subRL->_balance_factor = 0;

    //情况一
	if (bf == 1) {
		parent->_balance_factor = -1;
		subR->_balance_factor = 0;
	}
    //情况二
	else if (bf == -1) {
		parent->_balance_factor = 0;
		subR->_balance_factor = 1;
	}
    //情况三
	else if (bf == 0) {
		parent->_balance_factor = 0;
		subR->_balance_factor = 0;
	}

	else assert(false);
}

四、 AVL树的删除操作

因为AVL树也是二叉搜索树,可按照二叉搜索树的方式将节点删除,然后再更新平衡因子,只不
过与删除不同的时,删除节点后的平衡因子更新,最差情况下一直要调整到根节点的位置。

后续完善………………
 

五、 AVL树性能分析

AVL树是一棵绝对平衡的二叉搜索树,其要求每个节点的左右子树高度差的绝对值都不超过1
这样可以保证查询时高效的时间复杂度,即O(log_2 N)
但是若要对AVL树做一些结构修改的操作,性能非常低下。比如:插入时要维护其绝对平衡,旋转的次数比较多,更差的是在删除时,有可能一直要让旋转持续到根的位置
因此若需要一种查询高效且有序的数据结构,而且数据的个数为静态的(即不会改变),可以考虑AVL树,但一个结构经常修改,就不太适合

六、 完整代码

#include <utility>
#include <cstdlib>
#include <cassert>
#include <iostream>
using std::cout;
using std::endl;
using std::max;
using std::swap;
using std::pair;
using std::make_pair;

template<class K,class V>
struct AVLTreeNode {
	AVLTreeNode(const pair<K,V>& kv) :_parent(nullptr), _left(nullptr), _right(nullptr),_data(kv),_balance_factor(0) {}

	AVLTreeNode<K, V>* _parent;
	AVLTreeNode<K, V>* _left;
	AVLTreeNode<K, V>* _right;

	pair<K, V> _data;
	int _balance_factor;//平衡因子
};

template<class K, class V>
class AVLTree 
{
	typedef AVLTreeNode<K, V> TreeNode;
public:
	bool insert(const pair<K, V>& kv) {
		if (_root == nullptr) {
			_root = new TreeNode(kv);
			return true;
		}

		TreeNode* parent = nullptr;
		TreeNode* cur = _root;
		while (cur != nullptr) {
			if (kv.first > cur->_data.first) {
				parent = cur;
				cur = cur->_right;
			}
			else if (kv.first < cur->_data.first) {
				parent = cur;
				cur = cur->_left;
			}
			else return false;
		}
		cur = new TreeNode(kv);
		if (kv.first > parent->_data.first) {
			parent->_right = cur;
		}
		else { //kv.first < parent->_data.first)
			parent->_left = cur;
		}
		cur->_parent = parent;

		//控制平衡
		//更新平衡因子
		while (parent != nullptr){
			if (cur == parent->_right) ++parent->_balance_factor;
			else --parent->_balance_factor;

			if(parent->_balance_factor == 0) break;
			else if (abs(parent->_balance_factor) == 1) {
				cur = parent;
				parent = parent->_parent;
			}
			else if (abs(parent->_balance_factor) == 2) {
				//需要旋转
				if (parent->_balance_factor == 2 && cur->_balance_factor == 1) {
					rotate_left(parent);
				}
				else if (parent->_balance_factor == -2 && cur->_balance_factor == -1) {
					rotate_right(parent);
				}
				else if (parent->_balance_factor == -2 && cur->_balance_factor == 1) {
					rotate_left_right(parent);
				}
				else if (parent->_balance_factor == 2 && cur->_balance_factor == -1) {
					rotate_right_left(parent);
				}
				else assert(false);
				break;
			}
			else {
				assert(false);
			}
		}
		return true;
	}

	void inorder() {
		_inorder(_root);
	}
	bool IsBlance() {
		return _IsBlance(_root);
	}
private:
	void _inorder(TreeNode* root) {
		if (root == nullptr) {
			return;
		}
		_inorder(root->_left);
		cout << root->_data.first << " ";
		_inorder(root->_right);
	}

	bool _IsBlance(TreeNode* root) {
		if (root == nullptr) return true;

		int diff = Height(root->_right) - Height(root->_left);
		if (diff != root->_balance_factor) {
			cout << root->_data.first << "结点的平衡因子异常" << endl;
			return false;
		}

		return abs(diff) < 2 && _IsBlance(root->_left) && _IsBlance(root->_right);
	}
	int Height(TreeNode* root) {
		if (root == nullptr) return 0;
		return max(Height(root->_left),Height(root->_right)) + 1;
	}

	void rotate_left(TreeNode* parent) {
		TreeNode* subR = parent->_right;
		TreeNode* subRL = subR->_left;
		TreeNode* pparent = parent->_parent;

		parent->_right = subRL;
		if (subRL != nullptr) subRL->_parent = parent;
		subR->_left = parent;
		parent->_parent = subR;

		//解决根结点变换带来的问题
		if (_root == parent) {
			_root = subR;
			subR->_parent = nullptr;
		}
		else {
			if (pparent->_left == parent) pparent->_left = subR;
			else pparent->_right = subR;
			subR->_parent = pparent;
		}
		subR->_balance_factor = parent->_balance_factor = 0;
	}

	void rotate_right(TreeNode* parent) {
		TreeNode* subL = parent->_left;
		TreeNode* subLR = subL->_right;
		TreeNode* pparent = parent->_parent;

		parent->_left = subLR;
		if (subLR != nullptr) subLR->_parent = parent;
		subL->_right = parent;
		parent->_parent = subL;

		if (_root == parent) {
			_root = subL;
			subL->_parent = nullptr;
		}
		else {
			if (pparent->_left == parent) pparent->_left = subL;
			else pparent->_right = subL;
			subL->_parent = pparent;
		}
		subL->_balance_factor = parent->_balance_factor = 0;
	}

	void rotate_left_right(TreeNode* parent) {
		TreeNode* subL = parent->_left;
		TreeNode* subLR = subL->_right;
		int bf = subLR->_balance_factor;

		rotate_left(parent->_left);
		rotate_right(parent);

		//更新平衡因子
		subLR->_balance_factor = 0;
		if (bf == 1) {
			parent->_balance_factor = 0;
			subL->_balance_factor = -1;
		}
		else if (bf == -1) {
			parent->_balance_factor = 1;
			subL->_balance_factor = 0;
		}
		else if (bf == 0) {
			parent->_balance_factor = 0;
			subL->_balance_factor = 0;
		}
		else assert(false);
	}

	void rotate_right_left(TreeNode* parent) {
		TreeNode* subR = parent->_right;
		TreeNode* subRL = subR->_left;
		int bf = subRL->_balance_factor;

		rotate_right(parent->_right);
		rotate_left(parent);

		subRL->_balance_factor = 0;
		if (bf == 1) {
			parent->_balance_factor = -1;
			subR->_balance_factor = 0;
		}
		else if (bf == -1) {
			parent->_balance_factor = 0;
			subR->_balance_factor = 1;
		}
		else if (bf == 0) {
			parent->_balance_factor = 0;
			subR->_balance_factor = 0;
		}
		else assert(false);
	}
private:
	TreeNode* _root = nullptr;
};

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/17265.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

论文管理系统(准备工作)

目录 一、项目需求响应图 二、准备工作 2.1创建一个Spring Initializr项目 2.2后端架构 2.2.1 controller层 2.2.2 entity层 2.2.3 interceptors层 2.2.4 mapper层 2.2.5 serivice层 2.2.6 main运行 2.2.7 mappers文件 2.3配置 application.yml文件 2.4加入依赖 一、项…

Postman如何携带token——Bearer Token和Headers

目录一、使用场景二、设置Bearer Token1.设置你的环境变量2.项目集合设置认证方式及环境变量3.登录接口的脚本三、通过脚本设置Headers1.登录请求设置环境变量2.设置集合的发送请求脚本一、使用场景 现在许多项目都使用jwt来实现用户登录和数据权限&#xff0c;校验过用户的用…

C++ Builder XE关于AdvStringGrid对EXCEL母表快速分表,并批量插入sheet子表简单操作

如何快速将ECXEL母表快速批量生成多个子表分表&#xff0c;并且在表中插入sheet子表的简单操作&#xff1a; //AdvStringGrid2->SaveToXLS(filename,false);//生成新EXCEL表格 //AdvStringGrid2->SaveToXLSSheet(filename,Fname);//插入sheet子表 //-----------------…

一篇五分生信临床模型预测文章代码复现——Figure 4-6 临床模型构建(五)

之前讲过临床模型预测的专栏,但那只是基础版本,下面我们以自噬相关基因为例子,模仿一篇五分文章,将图和代码复现出来,学会本专栏课程,可以具备发一篇五分左右文章的水平: 本专栏目录如下: Figure 1:差异表达基因及预后基因筛选(图片仅供参考) Figure 2. 生存分析,…

XSS-labs靶场实战(五)——第12-14关

今天继续给大家介绍渗透测试相关知识&#xff0c;本文主要内容是XSS-labs靶场实战第12-14关。 免责声明&#xff1a; 本文所介绍的内容仅做学习交流使用&#xff0c;严禁利用文中技术进行非法行为&#xff0c;否则造成一切严重后果自负&#xff01; 再次强调&#xff1a;严禁对…

认识和使用容器

目录 前言 一、容器化背后的发展历史和概念 1.容器的抽象 容器的比喻 2.计算机领域的容器 容器是一种标准化的软件单元 二、容器和微服务架构 1.容器的作用 快速搭建开发环境 将运行环境和配置放在代码中并部署 使用docker-compose来模拟生产环境 使用docker镜像进…

在JVM 中进程与线程关系、介绍线程:程序计数器、本地方法栈、虚拟机栈

首先,我们要了解进程和线程的基本概念 进程 process 一个在内存中运行的应用程序。每个进程都有自己独立的一块内存空间,一个进程可以有多个线程,比如在Windows系统中,一个运行的*.exe应用程序就是一个进程。 线程 thread 进程中的一个执行任务(控制单元),负责当前进…

vue-element-admin依赖报错npm ERR! code 128 npm ERR! An unknown git error occurred

解决vue-element-admin安装报错 npm ERR code 128 npm ERR An unknown git error occurred npm 报错截图&#xff1a; 参考地址 先试一下&#xff1a;控制台输入&#xff1a; git config --global http.sslverify “false” git config --global url.“https://”.insteadOf …

几行代码实现用Python输出表情包

近几日在搞邮件自动发送&#xff0c;发现python原来可以发小表情&#xff01;而且操作很容易&#xff0c;但是发现现在的博文介绍的不是很全面&#xff08;或者我没搜出来……&#xff09;&#xff0c;因此在此补充一二。 1. python输出的表情样子 图里的表情包当然才是冰山一角…

【案例 5-3】 模拟用户登录

【案例介绍】 1.任务描述 在使用一些 APP 时&#xff0c;通常都需要填写用户名和密码。用户名和密码输入都正确才会登录成功&#xff0c;否则会提示用户名或密码错误。 本例要求编写一个程序&#xff0c;模拟用户登录。程序要求如下&#xff1a; &#xff08;1&#xff09; 用…

隐式转换这个概念你听说过没?

世界上最遥远的距离不是生与死&#xff0c;而是你亲手制造的BUG就在你眼前&#xff0c;你却怎么都找不到她。 目录 1、隐式转换是什么意思 1.1整型截断 1.2整形提升 2、char的取值范围 2.1有符号char取值范围 2.2无符号char取值范围 前言&#xff1a; 大家好&#xff0c;…

Java面向对象中阶(七)

面向对象中阶 1、包 2、访问修饰符 3、封装 4、继承 5、方法重写(override) 6、多态 7、Object类的常用方法 8、断点调试 1、包 包的本质&#xff1a; 实际上就是创建不同的文件夹来保存类文件 包的三大作用&#xff1a; 区分相同名字的类当类很多时&#xff0c;可以…

关于bios+mbr还原uefi+gpt的系统后,出现进不去系统的问题

问题&#xff1a; 重装系统后&#xff08;或还原Ghost系统后&#xff09;&#xff0c;出现如下的界面&#xff0c;应该是找不到引导所致&#xff0c;一直停留在这个界面&#xff0c;进不去系统 然后只能从pe界面选择从硬盘启动操作系统 这种通过pe系统&#xff0c;启动本地硬…

食品制造业SCM系统供应商管理模块提升企业采购管理效率,数字化升级供应链

民以食为天&#xff0c;随着人民生活水平的提高&#xff0c;人们对食品的购买趋向多品种、少数量、无污染的消费意愿&#xff0c;就要求食品企业抓住客户需求&#xff0c;及时提供安全健康的产品&#xff0c;因此畅通高效、精准智能的供应链就显得尤为重要。 食品行业供应链要…

深度学习与总结JVM专辑(一):基础介绍内存结构(图文+代码)

基础介绍&&内存结构前言基础介绍铺垫为什么要学习JVM什么是JVM为什么Java要在虚拟机里运行托管环境字节码文件Class文件的结构属性Java虚拟机具体是怎么运行Java字节码Java虚拟机的运行效率究竟如何JDK&JRE&JVM有什么区别内存管理程序计数器为什么要使用PC寄存…

01. Web 1~10

Web 1~10web1知识点题解web2知识点题解web3知识点题解web4知识点题解web5知识点题解web6知识点题解web7知识点题解web8知识点题解web9知识点题解web10知识点题解web1 知识点 查看网页源码&#xff1a;ctrl u 或 F12 开发注释未及时删除 题解 查看网页源码即可。 web2 知识…

[附源码]Python计算机毕业设计毕业设计管理系统

项目运行 环境配置&#xff1a; Pychram社区版 python3.7.7 Mysql5.7 HBuilderXlist pipNavicat11Djangonodejs。 项目技术&#xff1a; django python Vue 等等组成&#xff0c;B/S模式 pychram管理等等。 环境需要 1.运行环境&#xff1a;最好是python3.7.7&#xff0c;…

深入浅出PyTorch——主要模块和基础实战

一、深度学习模型训练流程“八股文” 1.1 机器学习的开发应用步骤&#xff1a; 数据搜集——>数据预处理——>特征工程——>划分训练集和测试集——>选择模型——>建立模型&#xff08;模型超参数设置&#xff09;——>设置损失函数——>进行训练以及预测…

WebRTC源码研究(4)web服务器工作原理和常用协议基础(转载)

前言 前面3篇博客分别对WebRTC框架的介绍&#xff0c;WebRTC源码目录&#xff0c;WebRTC的运行机制进行了介绍&#xff0c;接下来讲解一点关于服务器原理的知识。后面博客会写关于WebRTC服务器相关的开发&#xff0c;目前git上面有好多WebRTC相关的流媒体服务器的源码&#xff…

[附源码]java毕业设计农家乐点餐系统

项目运行 环境配置&#xff1a; Jdk1.8 Tomcat7.0 Mysql HBuilderX&#xff08;Webstorm也行&#xff09; Eclispe&#xff08;IntelliJ IDEA,Eclispe,MyEclispe,Sts都支持&#xff09;。 项目技术&#xff1a; SSM mybatis Maven Vue 等等组成&#xff0c;B/S模式 M…