AVL树
- 一、AVL树的概念
- 二、AVL的接口
- 2.1 插入
- 2.2 旋转
- 2.2.1 左单旋
- 2.2.2 右单旋
- 2.2.3 左右双旋
- 2.2.4 右左双旋
- 三、验证
- 四、源码
一、AVL树的概念
当我们用普通的搜索树插入数据的时候,如果插入的数据是有序的,那么就退化成了一个链表,搜索效率低下。
为了应对这种情况,就出现了AVL树(高度平衡二叉搜索树):
当向二叉搜索树中插入新结点后,如果能保证每个结点的左右子树高度之差的绝对值不超过1
AVL树的性质:
- 它的左右子树都是AVL树。
- 左右子树高度之差(简称平衡因子)的绝对值不超过1
平衡因子= 右子树高度-左子树高度
平衡因子是用来检测树的状态,如果平衡因子都在(-1, 0, 1)中,则没问题,反之则需要调整。
二、AVL的接口
AVL的节点定义:
template <class K, class V>
struct AVLNode
{
AVLNode(const pair<K, V>& kv)
: _kv(kv)
, _left(nullptr)
, _right(nullptr)
, _parent(nullptr)
, _bf(0)
{}
pair<K, V> _kv;
AVLNode<K, V>* _left;
AVLNode<K, V>* _right;
AVLNode<K, V>* _parent;
int _bf;// 平衡因子
};
2.1 插入
AVL的基本插入流程跟搜索树相似,但是AVL树多了一个平衡因子。
一旦插入新节点,就要往上更新平衡因子。
- 如果是在左边点插入,则平衡因子
--
- 如果是在右边点插入,则平衡因子
++
更新一个结点之后我们需要去进行判断,子树的高度是否发生了变化:
1️⃣ 当父节点的平衡因子变成0:说明原来是-1或1,那么也就是把矮的地方填平了,父节点所在树的高度不变,不需要继续更新。
2️⃣ 当父节点的平衡因子变成1或-1:说明原来是0,父节点所在树的高度发生变化,需要继续更新。
3️⃣ 当当父节点的平衡因子变成2或-2:违反规则,需要进行旋转处理。
所以我们可以利用parent节点(插入之前的叶子节点),从它开始往上更新。
bool insert(const pair<K, V>& kv)
{
if (_root == nullptr)
{
_root = new Node(kv);
return true;
}
Node* parent = nullptr;
Node* cur = _root;
while (cur)
{
if (kv.first < cur->_kv.first)
{
parent = cur;
cur = cur->_left;
}
else if (kv.first > cur->_kv.first)
{
parent = cur;
cur = cur->_right;
}
else return false;
}
cur = new Node(kv);
if (kv.first < parent->_kv.first)
{
parent->_left = cur;
}
else
{
parent->_right = cur;
}
cur->_parent = parent;
// 更新平衡因子
while (parent)
{
if (cur == parent->_left) parent->_bf--;
else parent->_bf++;
if (parent->_bf == 0) break;
else if (parent->_bf == 1 || parent->_bf == -1)
{
cur = parent;
parent = parent->_parent;
}
else if (parent->_bf == 2 || parent->_bf == -2)
{
// 旋转
if (parent->_bf == 2 && cur->_bf == 1)
{
// 左单旋
RotateL(parent);
}
else if (parent->_bf == -2 && cur->_bf == -1)
{
// 右单旋
RotateR(parent);
}
else if (parent->_bf == -2 && cur->_bf == 1)
{
// 左右双旋
RotateLR(parent);
}
else if (parent->_bf == 2 && cur->_bf == -1)
{
// 右左双旋
RotateRL(parent);
}
break;
}
else
{
cout << "结构出错" << endl;
assert(false);
}
}
return true;
}
2.2 旋转
为了保证每个结点的左右子树高度之差的绝对值不超过1,所以当平衡因子变为-2或2时,需要旋转来保持平衡。
旋转规则:
1️⃣ 让这颗子树左右高度差不超过1
2️⃣ 旋转过程中继续保持它是搜索树
3️⃣ 更新调整孩子节点的平衡因子
4️⃣ 让这颗子树的高度根插入前保持一致
2.2.1 左单旋
二叉树的结构有无数种情况,所以我们需要总结出抽象图来分析
解释:
a/b/c是高度为h的AVL树。
新节点插入较高右子树的右侧—右右:左单旋
左单旋的步骤:
1️⃣ 20的左边调整到10的右边
2️⃣ 10变成20的左边,20做根
3️⃣ 把平衡因子变为0
void RotateL(Node* parent)
{
Node* top = parent->_parent;
Node* right = parent->_right;
// 20的左边调整到10的右边
parent->_right = right->_left;
if (right->_left) right->_left->_parent = parent;
// 10变成20的左边,20做根
right->_left = parent;
parent->_parent = right;
if (top)// 子树
{
if (parent == top->_left) top->_left = right;
else top->_right = right;
right->_parent = top;
}
else// 完整的树
{
_root = right;
_root->_parent = nullptr;
}
// 更新平衡因子
parent->_bf = right->_bf = 0;
}
2.2.2 右单旋
新节点插入较高左子树的左侧—左左:右单旋
void RotateR(Node* parent)
{
Node* top = parent->_parent;
Node* left = parent->_left;
Node* leftR = left->_right;
parent->_left = leftR;
if (leftR) leftR->_parent = parent;
left->_right = parent;
parent->_parent = left;
if (top)
{
if (parent == top->_left) top->_left = left;
else top->_right = left;
left->_parent = top;
}
else
{
_root = left;
_root->_parent = nullptr;
}
parent->_bf = left->_bf = 0;
}
2.2.3 左右双旋
新节点插入较高左子树的右侧—左右:先左单旋再右单旋
我们看到上面的单旋,我们会想,如果是这么插入呢?
其实这个图可以转化为:
先以10为轴进行左单旋,这样就把“折线”变成了直线,在以20为轴进行右单旋。
这里就要注意平衡因子的更新
15的平衡因子为0
但是其他两个会有三个不同的情况:
1️⃣ 当right的平衡因子为-1时(插入在b),双旋结束后parent、left、right的平衡因子分别更新为1、0、0
2️⃣ 当right的平衡因子为1时(插入在c),双旋结束后parent、left、right的平衡因子分别更新为0、-1、0
3️⃣ 当right的平衡因子为0时,双旋后parent、left、right的平衡因子分别更新为0、0、0
所以在旋转前要先进行判断在哪插入(通过平衡因子),旋转后手动更新即可。
void RotateLR(Node* parent)
{
Node* left = parent->_left;
Node* right = left->_right;
int bf = right->_bf;// 提前记录
RotateL(parent->_left);
RotateR(parent);
if (bf == -1)// 左子树新增
{
left->_bf = 0;
right->_bf = 0;
parent->_bf = 1;
}
else if (bf == 1)// 右子树新增
{
left->_bf = -1;
right->_bf = 0;
parent->_bf = 0;
}
else if (bf == 0)// 自己就是新增
{
left->_bf = right->_bf = parent->_bf = 0;
}
else assert(false);
}
2.2.4 右左双旋
void RotateRL(Node* parent)
{
Node* right = parent->_right;
Node* left = right->_left;
int bf = left->_bf;
RotateR(right);
RotateL(parent);
if (bf == -1)
{
parent->_bf = 0;
left->_bf = 0;
right->_bf = 1;
}
else if (bf == 1)
{
right->_bf = 0;
left->_bf = 0;
parent->_bf = -1;
}
else if (bf == 0)
{
left->_bf = right->_bf = parent->_bf = 0;
}
else assert(false);
}
三、验证
为了验证是否为二叉搜索树,我们可以先写一个中序遍历
void _Inorder(Node* root)
{
if (root == nullptr)
return;
_Inorder(root->_left);
cout << root->_kv.first << ":" << root->_kv.second << endl;
_Inorder(root->_right);
}
void Inorder()
{
_Inorder(_root);
}
为了验证是否为AVL树,我们要让每个节点的左右子树高度的绝对值差小于等于1。
int Height(Node* root)
{
if (!root)
{
return 0;
}
int lh = Height(root->_left) + 1;
int rh = Height(root->_right) + 1;
return max(lh, rh);
}
bool IsBalance(Node* root)
{
if (!root)
{
return true;
}
int lh = Height(root->_left);
int rh = Height(root->_right);
if (rh - lh != root->_bf)
{
cout << root->_kv.first << ":";
cout << root->_bf << ":";
cout << "平衡因子出错" << endl;
return false;
}
if (abs(rh - lh) > 1)
{
return false;
}
return IsBalance(root->_left) && IsBalance(root->_right);
}
bool IsBalance()
{
return IsBalance(_root);
}
我们可以用大量的随机值来测定:
void test()
{
const int N = 100000;
AVLTree<int, int> tt;
srand(time(0));
for (int i = 0; i < N; i++)
{
int x = rand();
tt.insert(make_pair(x, x));
}
//tt.Inorder();
cout << tt.IsBalance() << endl;
}
四、源码
#pragma once
#include <iostream>
#include <string>
#include <cassert>
#include <cstdlib>
using namespace std;
template <class K, class V>
struct AVLNode
{
AVLNode(const pair<K, V>& kv)
: _kv(kv)
, _left(nullptr)
, _right(nullptr)
, _parent(nullptr)
, _bf(0)
{}
pair<K, V> _kv;
AVLNode<K, V>* _left;
AVLNode<K, V>* _right;
AVLNode<K, V>* _parent;
int _bf;// 平衡因子
};
template <class K, class V>
class AVLTree
{
typedef AVLNode<K, V> Node;
public:
bool insert(const pair<K, V>& kv)
{
if (_root == nullptr)
{
_root = new Node(kv);
return true;
}
Node* parent = nullptr;
Node* cur = _root;
while (cur)
{
if (kv.first < cur->_kv.first)
{
parent = cur;
cur = cur->_left;
}
else if (kv.first > cur->_kv.first)
{
parent = cur;
cur = cur->_right;
}
else return false;
}
cur = new Node(kv);
if (kv.first < parent->_kv.first)
{
parent->_left = cur;
}
else
{
parent->_right = cur;
}
cur->_parent = parent;
// 更新平衡因子
while (parent)
{
if (cur == parent->_left) parent->_bf--;
else parent->_bf++;
if (parent->_bf == 0) break;
else if (parent->_bf == 1 || parent->_bf == -1)
{
cur = parent;
parent = parent->_parent;
}
else if (parent->_bf == 2 || parent->_bf == -2)
{
// 旋转
if (parent->_bf == 2 && cur->_bf == 1)
{
// 左单旋
RotateL(parent);
}
else if (parent->_bf == -2 && cur->_bf == -1)
{
// 右单旋
RotateR(parent);
}
else if (parent->_bf == -2 && cur->_bf == 1)
{
// 左右双旋
RotateLR(parent);
}
else if (parent->_bf == 2 && cur->_bf == -1)
{
// 右左双旋
RotateRL(parent);
}
break;
}
else
{
cout << "结构出错" << endl;
assert(false);
}
}
return true;
}
void RotateL(Node* parent)
{
Node* top = parent->_parent;
Node* right = parent->_right;
// 20的左边调整到10的右边
parent->_right = right->_left;
if (right->_left) right->_left->_parent = parent;
// 10变成20的左边,20做根
right->_left = parent;
parent->_parent = right;
if (top)// 子树
{
if (parent == top->_left) top->_left = right;
else top->_right = right;
right->_parent = top;
}
else// 完整的树
{
_root = right;
_root->_parent = nullptr;
}
// 更新平衡因子
parent->_bf = right->_bf = 0;
}
void RotateR(Node* parent)
{
Node* top = parent->_parent;
Node* left = parent->_left;
Node* leftR = left->_right;
parent->_left = leftR;
if (leftR) leftR->_parent = parent;
left->_right = parent;
parent->_parent = left;
if (top)
{
if (parent == top->_left) top->_left = left;
else top->_right = left;
left->_parent = top;
}
else
{
_root = left;
_root->_parent = nullptr;
}
parent->_bf = left->_bf = 0;
}
void RotateLR(Node* parent)
{
Node* left = parent->_left;
Node* right = left->_right;
int bf = right->_bf;// 提前记录
RotateL(left);
RotateR(parent);
if (bf == -1)// 左子树新增
{
left->_bf = 0;
right->_bf = 0;
parent->_bf = 1;
}
else if (bf == 1)// 右子树新增
{
left->_bf = -1;
right->_bf = 0;
parent->_bf = 0;
}
else if (bf == 0)// 自己就是新增
{
left->_bf = right->_bf = parent->_bf = 0;
}
else assert(false);
}
void RotateRL(Node* parent)
{
Node* right = parent->_right;
Node* left = right->_left;
int bf = left->_bf;
RotateR(right);
RotateL(parent);
if (bf == -1)
{
parent->_bf = 0;
left->_bf = 0;
right->_bf = 1;
}
else if (bf == 1)
{
right->_bf = 0;
left->_bf = 0;
parent->_bf = -1;
}
else if (bf == 0)
{
left->_bf = right->_bf = parent->_bf = 0;
}
else assert(false);
}
void _Inorder(Node* root)
{
if (root == nullptr)
return;
_Inorder(root->_left);
cout << root->_kv.first << "<=>" << root->_kv.second << endl;
_Inorder(root->_right);
}
void Inorder()
{
_Inorder(_root);
}
int Height(Node* root)
{
if (!root)
{
return 0;
}
int lh = Height(root->_left) + 1;
int rh = Height(root->_right) + 1;
return max(lh, rh);
}
bool IsBalance(Node* root)
{
if (!root)
{
return true;
}
int lh = Height(root->_left);
int rh = Height(root->_right);
if (rh - lh != root->_bf)
{
cout << root->_kv.first << ":";
cout << root->_bf << ":";
cout << "平衡因子出错" << endl;
return false;
}
if (abs(rh - lh) > 1)
{
return false;
}
return IsBalance(root->_left) && IsBalance(root->_right);
}
bool IsBalance()
{
return IsBalance(_root);
}
private:
Node* _root = nullptr;
};
void test()
{
const int N = 100000;
AVLTree<int, int> tt;
srand(time(0));
for (int i = 0; i < N; i++)
{
int x = rand();
tt.insert(make_pair(x, x));
}
//tt.Inorder();
cout << tt.IsBalance() << endl;
}