nature
Le concept de facteur d'équilibre est introduit sur la base de l'arbre BST, qui exige que la différence de hauteur entre les sous-arbres gauche et droit de tout nœud ne dépasse pas 1
Quatre situations qui nécessitent une rotation
- L'arbre de l'enfant gauche de l'enfant gauche est trop grand : droitier
- L'arbre de l'enfant droit de l'enfant droit est trop grand : gaucher
- Le sous-arbre droit de l'enfant gauche est trop haut : faites d'abord pivoter l'enfant gauche vers la gauche, puis faites pivoter le nœud actuel vers la droite (équilibre à gauche).
- Le sous-arbre gauche de l'enfant droit est trop haut : tournez d'abord à droite vers l'enfant droit, puis tournez à gauche vers le nœud actuel (équilibre à droite)
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
// 定义节点类型
template<typename T>
struct Node {
Node(T data = T()) : data_(data), left_(nullptr), right_(nullptr), height_(1) {}
T data_;
Node* left_;
Node* right_;
int height_; // 记录节点的高度
};
// AVL树
template<typename T>
class AVLTree {
public:
AVLTree() : root_(nullptr) {}
// 插入
void insert(const T& val) {
root_ = insert(root_,val);
}
// 删除
void remove(const T& val) {
root_ = remove(root_,val);
}
private:
Node<T>* root_; // 根节点
// 返回节点的高度
int height(Node<T> *node) {
return node == nullptr ? 0 : node->height_;
}
// 右旋
Node<T>* rightRotate(Node<T>* node);
// 左旋
Node<T>* leftRotate(Node<T>* node);
// 左平衡
Node<T>* leftBalance(Node<T>* node);
// 右平衡
Node<T>* rightBalance(Node<T>* node);
// 插入
Node<T>* insert(Node<T>* node, const T& val);
// 删除
Node<T>* remove(Node<T>* node, const T& val);
};
// 右旋
template<typename T>
Node<T>* AVLTree<T>::rightRotate(Node<T>* node) {
// 节点旋转
Node<T>* child = node->left_;
node->left_ = child->right_;
child->right_ = node;
// 高度更新
node->height_ = max(height(node->left_), height(node->right_)) + 1;
child->height_ = max(height(child->left_), height(child->right_)) + 1;
// 返回旋转后的子树的新根节点
return child;
}
// 左旋
template<typename T>
Node<T>* AVLTree<T>::leftRotate(Node<T>* node) {
// 节点旋转
Node<T> *child = node->left_;
node->right_ = child->left_;
child->left_ = node;
// 高度更新
node->height_ = max(height(node->left_), height(node->right_)) + 1;
child->height_ = max(height(child->left_), height(child->right_)) + 1;
// 返回旋转后的子树的新根节点
return child;
}
// 左平衡 先对node的左子树左旋,再对node右旋
template<typename T>
Node<T>* AVLTree<T>::leftBalance(Node<T> *node) {
node->left_ = leftRotate(node->left_);
return rightRotate(node);
}
// 右平衡 先对node的右子树右旋,再对node左旋
template<typename T>
Node<T>* AVLTree<T>::rightBalance(Node<T> *node) {
node->right_ = rightRotate(node->right_);
return leftRotate(node);
}
// 插入
template<typename T>
Node<T>* AVLTree<T>::insert(Node<T> *node, const T &val) {
// 递归结束 找到插入的位置
if (node == nullptr) return new Node<T>(val);
if (node->data_ > val) {
node->left_ = insert(node->left_,val);
// 判断是否失衡
if (height(node->left_) - height(node->right_) > 1) {
if (height(node->left_->left_) >= height(node->left_->right_)) {
// 左孩子的左子树太高
node = rightRotate(node);
} else {
// 左孩子的右子树太高
node = leftBalance(node);
}
}
} else if (node->data_ < val) {
node->right_ = insert(node->right_,val);
if (height(node->right_) - height(node->left_) > 1) {
if (height(node->right_->right_) >= height(node->right_->left_)) {
node = leftRotate(node);
} else {
node = rightBalance(node);
}
}
} else {
// 找到相同节点 不需要向下递归 直接向上回溯
}
// 因为子树添加了新的节点 所以在递归的时候需要更新节点高度
node->height_ = max(height(node->left_), height(node->right_)) + 1;
return node;
}
// 删除操作 从叶子节点中选出一个节点 进行替换
template<typename T>
Node<T>* AVLTree<T>::remove(Node<T> *node, const T &val) {
if (node == nullptr) {
return nullptr;
}
if (node->data_ > val) {
node->left_ = remove(node->left_,val);
if (height(node->right_) - height(node->left_) > 1) {
if (height(node->right_->right_) >= height(node->right_->left_)) {
node = leftRotate(node);
} else {
node = rightBalance(node);
}
}
} else if (node->data_ < val) {
node->right_ = remove(node->right_,val);
if (height(node->left_) - height(node->right_) > 1) {
if (height(node->left_->left_) >= height(node->left_->right_)) {
node = rightRotate(node);
} else {
node = leftBalance(node);
}
}
} else {
// 找到节点
// 如果有两个孩子
if (node->left_ != nullptr && node->right_ != nullptr) {
// 谁高删谁的节点
if (height(node->left_) >= height(node->right_)) {
Node<T>* pre = node->left_;
while (pre->right_ != nullptr) {
pre = pre->right_;
}
node->data_ = pre->data_;
node->left_ = remove(node->left_,pre->data_);
} else {
Node<T>* pre = node->right_;
while (pre->left_ != nullptr) {
pre = pre->left_;
}
node->data_ = pre->data_;
node->right_ = remove(node->right_,pre->data_);
}
} else {
// 如果只有一个孩子
if (node->left_ != nullptr) {
Node<T>* left = node->left_;
delete node;
return left;
} else if (node->right_ != nullptr) {
Node<T>* right = node->right_;
delete node;
return right;
} else {
delete node;
return nullptr;
}
}
}
// 更新节点高度
node->height_ = max(height(node->left_), height(node->right_)) + 1;
return node;
}
analyse de performance
- L'insertion d'un nœud dans un arbre AVL nécessite au plus deux rotations pour rétablir l'équilibre
L'insertion d'un nœud entraînera l'augmentation de la hauteur du sous-arbre où se trouve le nœud de 1, mais la rotation réduira le sous-arbre où se trouve le nouveau nœud de 1, donc l'insertion d'un nœud dans un arbre AVL ne nécessite que deux rotations au plus
- La suppression de l'arborescence AVL d'un nœud nécessite au plus O(logN) rotations pour rétablir l'équilibre
La suppression d'un nœud réduira le sous-arbre où le nœud est situé de 1, et la rotation réduira le sous-arbre où le nœud est situé de 1, donc dans le pire des cas, des rotations O (logN) sont nécessaires
Après la suppression du nœud X, le facteur d'équilibre de R4 devient -2 et R4 est gaucher ; le facteur d'équilibre de R3 devient 2 et R3 est droitier ; le facteur d'équilibre de R2 devient -2 et R2 est gauche- gaucher ; le facteur d'équilibre de R1 devient 2, et R1 Droitier
Lorsque le facteur d'équilibre entre le nœud racine et le nœud parent du nœud à supprimer est -1 et +1 alternativement, une fois le nœud supprimé et la rotation déclenchée, des rotations de connexion sont nécessaires