Merge pull request #68 from mgisr/master
Create 'avl_tree' and modify 'TreeNode.hpp' to support 'avl_tree'pull/88/head
@ -0,0 +1,228 @@
|
||||
/*
|
||||
* File: avl_tree.cpp
|
||||
* Created Time: 2022-12-2
|
||||
* Author: mgisr (maguagua0706@gmail.com)
|
||||
*/
|
||||
|
||||
#include "../include/include.hpp"
|
||||
|
||||
class AvlTree {
|
||||
private:
|
||||
TreeNode *root{};
|
||||
static bool isBalance(const TreeNode *p);
|
||||
static int getBalanceFactor(const TreeNode *p);
|
||||
static void updateHeight(TreeNode *p);
|
||||
void fixBalance(TreeNode *p);
|
||||
static bool isLeftChild(const TreeNode *p);
|
||||
static TreeNode *&fromParentTo(TreeNode *node);
|
||||
public:
|
||||
AvlTree() = default;
|
||||
AvlTree(const AvlTree &p) = default;
|
||||
const TreeNode *search(int val);
|
||||
bool insert(int val);
|
||||
bool remove(int val);
|
||||
void printTree();
|
||||
};
|
||||
|
||||
// 判断该结点是否平衡
|
||||
bool AvlTree::isBalance(const TreeNode *p) {
|
||||
int balance_factor = getBalanceFactor(p);
|
||||
if (-1 <= balance_factor && balance_factor <= 1) { return true; }
|
||||
else { return false; }
|
||||
}
|
||||
|
||||
// 获取当前结点的平衡因子
|
||||
int AvlTree::getBalanceFactor(const TreeNode *p) {
|
||||
if (p->left == nullptr && p->right == nullptr) { return 0; }
|
||||
else if (p->left == nullptr) { return (-1 - p->right->height); }
|
||||
else if (p->right == nullptr) { return p->left->height + 1; }
|
||||
else { return p->left->height - p->right->height; }
|
||||
}
|
||||
|
||||
// 更新结点高度
|
||||
void AvlTree::updateHeight(TreeNode *p) {
|
||||
if (p->left == nullptr && p->right == nullptr) { p->height = 0; }
|
||||
else if (p->left == nullptr) { p->height = p->right->height + 1; }
|
||||
else if (p->right == nullptr) { p->height = p->left->height + 1; }
|
||||
else { p->height = std::max(p->left->height, p->right->height) + 1; }
|
||||
}
|
||||
|
||||
void AvlTree::fixBalance(TreeNode *p) {
|
||||
// 左旋操作
|
||||
auto rotate_left = [&](TreeNode *node) -> TreeNode * {
|
||||
TreeNode *temp = node->right;
|
||||
temp->parent = p->parent;
|
||||
node->right = temp->left;
|
||||
if (temp->left != nullptr) {
|
||||
temp->left->parent = node;
|
||||
}
|
||||
temp->left = node;
|
||||
node->parent = temp;
|
||||
updateHeight(node);
|
||||
updateHeight(temp);
|
||||
return temp;
|
||||
};
|
||||
// 右旋操作
|
||||
auto rotate_right = [&](TreeNode *node) -> TreeNode * {
|
||||
TreeNode *temp = node->left;
|
||||
temp->parent = p->parent;
|
||||
node->left = temp->right;
|
||||
if (temp->right != nullptr) {
|
||||
temp->right->parent = node;
|
||||
}
|
||||
temp->right = node;
|
||||
node->parent = temp;
|
||||
updateHeight(node);
|
||||
updateHeight(temp);
|
||||
return temp;
|
||||
};
|
||||
// 根据规则选取旋转方式
|
||||
if (getBalanceFactor(p) > 1) {
|
||||
if (getBalanceFactor(p->left) > 0) {
|
||||
if (p->parent == nullptr) { root = rotate_right(p); }
|
||||
else { fromParentTo(p) = rotate_right(p); }
|
||||
} else {
|
||||
p->left = rotate_left(p->left);
|
||||
if (p->parent == nullptr) { root = rotate_right(p); }
|
||||
else { fromParentTo(p) = rotate_right(p); }
|
||||
}
|
||||
} else {
|
||||
if (getBalanceFactor(p->right) < 0) {
|
||||
if (p->parent == nullptr) { root = rotate_left(p); }
|
||||
else { fromParentTo(p) = rotate_left(p); }
|
||||
} else {
|
||||
p->right = rotate_right(p->right);
|
||||
if (p->parent == nullptr) { root = rotate_left(p); }
|
||||
else { fromParentTo(p) = rotate_left(p); }
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 判断当前结点是否为其父节点的左孩子
|
||||
bool AvlTree::isLeftChild(const TreeNode *p) {
|
||||
if (p->parent == nullptr) { return false; }
|
||||
return (p->parent->left == p);
|
||||
}
|
||||
|
||||
// 返回父节点指向当前结点指针的引用
|
||||
TreeNode *&AvlTree::fromParentTo(TreeNode *node) {
|
||||
if (isLeftChild(node)) { return node->parent->left; }
|
||||
else { return node->parent->right; }
|
||||
}
|
||||
|
||||
const TreeNode *AvlTree::search(int val) {
|
||||
TreeNode *p = root;
|
||||
while (p != nullptr) {
|
||||
if (p->val == val) { return p; }
|
||||
else if (p->val > val) { p = p->left; }
|
||||
else { p = p->right; }
|
||||
}
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
bool AvlTree::insert(int val) {
|
||||
TreeNode *p = root;
|
||||
if (p == nullptr) {
|
||||
root = new TreeNode(val);
|
||||
return true;
|
||||
}
|
||||
for (;;) {
|
||||
if (p->val == val) { return false; }
|
||||
else if (p->val > val) {
|
||||
if (p->left == nullptr) {
|
||||
p->left = new TreeNode(val, p);
|
||||
break;
|
||||
} else {
|
||||
p = p->left;
|
||||
}
|
||||
} else {
|
||||
if (p->right == nullptr) {
|
||||
p->right = new TreeNode(val, p);
|
||||
break;
|
||||
} else {
|
||||
p = p->right;
|
||||
}
|
||||
}
|
||||
}
|
||||
for (; p != nullptr; p = p->parent) {
|
||||
if (!isBalance(p)) {
|
||||
fixBalance(p);
|
||||
break;
|
||||
} else { updateHeight(p); }
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool AvlTree::remove(int val) {
|
||||
TreeNode *p = root;
|
||||
if (p == nullptr) { return false; }
|
||||
while (p != nullptr) {
|
||||
if (p->val == val) {
|
||||
TreeNode *real_delete_node = p;
|
||||
TreeNode *next_node;
|
||||
if (p->left == nullptr) {
|
||||
next_node = p->right;
|
||||
if (p->parent == nullptr) { root = next_node; }
|
||||
else { fromParentTo(p) = next_node; }
|
||||
} else if (p->right == nullptr) {
|
||||
next_node = p->left;
|
||||
if (p->parent == nullptr) { root = next_node; }
|
||||
else { fromParentTo(p) = next_node; }
|
||||
} else {
|
||||
while (real_delete_node->left != nullptr) {
|
||||
real_delete_node = real_delete_node->left;
|
||||
}
|
||||
std::swap(p->val, real_delete_node->val);
|
||||
next_node = real_delete_node->right;
|
||||
if (real_delete_node->parent == p) { p->right = next_node; }
|
||||
else { real_delete_node->parent->left = next_node; }
|
||||
}
|
||||
if (next_node != nullptr) {
|
||||
next_node->parent = real_delete_node->parent;
|
||||
}
|
||||
for (p = real_delete_node; p != nullptr; p = p->parent) {
|
||||
if (!isBalance(p)) { fixBalance(p); }
|
||||
updateHeight(p);
|
||||
}
|
||||
delete real_delete_node;
|
||||
return true;
|
||||
} else if (p->val > val) {
|
||||
p = p->left;
|
||||
} else {
|
||||
p = p->right;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
void inOrder(const TreeNode *root) {
|
||||
if (root == nullptr) return;
|
||||
inOrder(root->left);
|
||||
cout << root->val << ' ';
|
||||
inOrder(root->right);
|
||||
}
|
||||
|
||||
void AvlTree::printTree() {
|
||||
inOrder(root);
|
||||
cout << endl;
|
||||
}
|
||||
|
||||
int main() {
|
||||
AvlTree tree = AvlTree();
|
||||
// tree.insert(13);
|
||||
// tree.insert(24);
|
||||
// tree.insert(37);
|
||||
// tree.insert(90);
|
||||
// tree.insert(53);
|
||||
|
||||
tree.insert(53);
|
||||
tree.insert(90);
|
||||
tree.insert(37);
|
||||
tree.insert(24);
|
||||
tree.insert(13);
|
||||
tree.remove(90);
|
||||
tree.printTree();
|
||||
const TreeNode *p = tree.search(37);
|
||||
cout << p->val;
|
||||
return 0;
|
||||
}
|
After Width: | Height: | Size: 62 KiB |
After Width: | Height: | Size: 57 KiB |
After Width: | Height: | Size: 78 KiB |
After Width: | Height: | Size: 86 KiB |
After Width: | Height: | Size: 82 KiB |
After Width: | Height: | Size: 42 KiB |
After Width: | Height: | Size: 56 KiB |
After Width: | Height: | Size: 59 KiB |
After Width: | Height: | Size: 67 KiB |
After Width: | Height: | Size: 89 KiB |
After Width: | Height: | Size: 69 KiB |