编程语言
首页 > 编程语言> > AVL平衡二叉树C++代码实现

AVL平衡二叉树C++代码实现

作者:互联网

总结

  1. 什么是平衡二叉树

    • 基于二叉排序树
    • 左右子树的深度之差的绝对值不超过1
    • 左右子树都是平衡二叉树
  2. 为什么要修改二叉排序树为平衡二叉树:因为查找二叉树的比较次数和层数有关

  3. 在构造二叉排序树的过程中,会出现四种失衡现象

  4. 如何进行调整:找到最小不平衡子树,将其调平衡

    • 最小不平衡子树:离插入节点最近且平衡因子绝对值超过1的结点,以这个结点为根节点的子树
  5. LL型:右旋,原本橙点是root,右旋后,绿点是root,橙点为绿点的right

    • 注意:绿点的right可能有结点,所以要用橙点的left接上
    • 以上图第三个为例
  6. RR型:左旋,原本橙点是root,左旋后,绿点是root,橙点为绿点的left

    • 注意:绿点的left可能有结点,所以要用橙点的right接上
    • 以上图第三个为例
  7. LR型:双旋,先对绿点(LL中的橙点)和蓝点(LL中的绿点)进行RR左旋,在对橙点进行LL右旋

    • 解释:左旋使LR型变为LL型,右旋使LL型平衡
    • 注意:对两个点也能进行RR左旋

  8. RL型:和LR型对称操作

代码思路

  1. 注意过程中要对树进行操作的函数的参数一定要传入指针的引用
  2. 递归插入结点insert
    • 如果curr为空,即找到最终位置,申请内存新建节点
    • 如果传入参数num小于当前结点的data,递归进入左子树
    • 如果传入参数num大于当前节点的data,递归进入右子树
  3. 插入结点后,就要退出递归,进行平衡因子的更新
    • 通过getHeight即可得到左子树和右子树的高度
    • 若平衡因子<1,继续退出递归
    • 若平衡因子>1,即不平衡,就要判断不平衡的类型了
      • 由第一张图可得,RX型橙点-2,LX型橙点2
      • RL型绿点1,RR型绿点-1,LR型绿点-1,LL型绿点1
      • 每个类型都有自己的特征,进入if判断
  4. 右旋要注意绿点的right,左旋要注意绿点的left
  5. 插入节点后和调平衡后,要更新改变过的结点的高度,绿点和橙点

代码

#include <iostream>
#include <vector>
using namespace std;
struct treeNode {
	int data;
	int height;
	treeNode* left, * right;
};
//中序遍历
void inorderTra(treeNode* curr) {
	if (!curr) return;
	inorderTra(curr->left);
	cout << curr->data << " ";
	inorderTra(curr->right);
}
//递归获取高度
int getHeight(treeNode* curr) {
	if (!curr) return -1;
	return curr->height;
}
//LL型,右旋单旋
treeNode* LL(treeNode*& curr) {
	treeNode* leftNode = curr->left;
	curr->left = leftNode->right;
	leftNode->right = curr;
	curr = leftNode;
	curr->height = max(getHeight(curr->left), getHeight(curr->right)) + 1;
	//可能为空,但不为空要重新判断高度
	if(curr->left) curr->left->height = max(getHeight(curr->left->left), getHeight(curr->left->right)) + 1;
	return curr;
}
//RR型,左旋单旋
treeNode* RR(treeNode*& curr) {
	treeNode* rightNode = curr->right;
	curr->right = rightNode->left;
	rightNode->left = curr;
	curr = rightNode;
	curr->height = max(getHeight(curr->left), getHeight(curr->right)) + 1;
	//可能为空,但不为空要重新判断高度
	if(curr->right) curr->right->height = max(getHeight(curr->right->left), getHeight(curr->right->right)) + 1;
	return curr;
}
//LR型,先对左子树右旋再左旋
treeNode* LR(treeNode*& curr) {
	RR(curr->left);
	return LL(curr);
}
//RL型,先对右子树左旋再右旋
treeNode* RL(treeNode*& curr) {
	LL(curr->right);
	return RR(curr);
}
//插入叶子结点,递归
//注意使用指针的引用
void insert(int num, treeNode*& curr) {
	//如果为空,即找到了叶子结点的位置,分配空间
	if (!curr) {
		curr = new treeNode();
		curr->data = num;
		curr->left = NULL;
		curr->right = NULL;
	}
	//如果数字比当前结点的值小,即进入当前结点的左子树继续判断
	else if (num < curr->data) {
		insert(num, curr->left);
		//递归出来后,要判断平衡因子,决定是否调平衡
		if (getHeight(curr->left) - getHeight(curr->right) == 2) {
			//如果上面递归是进入了左子树的左子树,则是LL型
			if (num < curr->left->data) curr = LL(curr);
			//如果进入左子树的右子树,则是LR型
			else curr = LR(curr);
		}
	}
	//如果数字比当前节点的值大,即进入当前节点的右子树继续判断
	else if (num > curr->data) {
		insert(num, curr->right);
		//递归出来后,要判断平衡因子,决定是否调平衡
		if (getHeight(curr->left) - getHeight(curr->right) == -2) {
			//如果上面递归是进入了右子树的右子树,则是RR型
			if (num > curr->right->data) curr = RR(curr);
			//如果进入右子树的左子树,则是RL型
			else curr = RL(curr);
		}
	}
	curr->height = max(getHeight(curr->left), getHeight(curr->right)) + 1;
}
void createTree(vector<int> v, treeNode*&pRoot) {
	for (int i = 0; i < v.size(); i++) {
		insert(v[i], pRoot);
	}
	//中序遍历,平衡二叉树是特殊的二叉排序树
	inorderTra(pRoot);
}
int main() {
	vector<int> v = { 49,38,65,97,76,13,27 };
	treeNode* pRoot = NULL;
	createTree(v,pRoot);
	return 0;
}

结果

标签:right,curr,绿点,getHeight,C++,AVL,二叉树,treeNode,left
来源: https://www.cnblogs.com/wasi-991017/p/13824039.html