#pragma once
#include<iostream>
#include<assert.h>
#include<vector>
template<class K,class V>
struct AVLtreeNode
{
pair<K,V> _kv;
AVLtreeNode<K, V>* _left;
AVLtreeNode<K, V>* _right;
AVLtreeNode<K, V>* _parent;//将父亲结点也连接上
//平衡因子
int _bf; //balance factor
AVLtreeNode(const pair<K,V>&kv)
:_kv(kv)
,_left(nullptr)
,_right(nullptr)
,_parent(nullptr)
,_bf(0)
{}
};
template <class K,class V>
class AVLtree
{
typedef AVLtreeNode<K, V> Node;
public:
bool insert(const pair<K, V>& kv)
{
if (_root == nullptr)
{
_root = new Node(kv);
return true;
}
Node* cur = _root;
Node* parent = nullptr;
//先遍历,找到空后才能插入
while(cur)
{
if (cur->_kv.first < kv.first)
{
parent = cur;//记录上一个节点
cur = cur->_right;
}
else if (cur->_kv.first > kv.first)
{
parent = cur;
cur = cur->_left;
}
else
{
return false;
}
}
//cur为空后
cur = new Node(kv);
if (kv.first > parent->_kv.first)
{
parent->_right = cur;
}
else if (kv.first < parent->_kv.first)
{
parent->_left = cur;
}
//连接一下父子节点,插入一个链接一个
cur->_parent = parent;
//更新平衡因子
//插入后parent就是新插入cur的父亲节点,_bf是右树高度减左树高度
//插入到左边就是_bf--,右边_bf++
while (parent)//因为_root的parent是空
{
if (cur == parent->_left)
{
parent->_bf--;
}
else if (cur == parent->_right)
{
parent->_bf++;
}
//等于0时,就不用再往上更新了
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)
{
RotateRL(parent);
}
//先左单旋,再右旋
else if (parent->_bf == -2 && cur->_bf == 1)
{
RotateLR(parent);
}
}
else
{
assert(false);
}
}
return true;
}
//旋转Rotate,左旋转
//左旋转,平衡因子为-2或者2时且右边树高的情况,需要旋转
//parent是平衡因子为2或者-2的节点,cur是右孩子
void RotateL(Node* parent)
{
Node* cur = parent->_right;
Node* curleft = cur->_left;
//核心操作
parent->_right = curleft;
if (curleft)
{
curleft->_parent = parent;
}
cur->_left = parent;
//再与上个节点连接上,因为左旋的可能是子树,也有可能是整个树
//ppnode是parent的父节点
//连接时,也需要判断链接左还是右
Node* ppnode = parent->_parent;
parent->_parent = cur;
if (parent == _root)
{
_root = cur;
cur->_parent = nullptr;
}
else
{
//判断链接左还是右,原来连接的是左是右就继续怎么连接,不用判断大小
if (ppnode->_left == parent)
{
ppnode->_left = cur;
}
else if (ppnode->_right == parent)
{
ppnode->_right = cur;
}
cur->_parent = ppnode;
}
//旋转完成后,更新平衡因子
cur->_bf = parent->_bf = 0;
}
//右旋
void RotateR(Node*parent)
{
Node* cur = parent->_left;
Node* curright = cur->_right;
parent->_left = curright;
if (curright != nullptr)
{
curright->_parent = parent;
}
cur->_right = parent;
//先存上
Node* ppnode = parent->_parent;
parent->_parent = cur;
//(ppnode==nullptr)
if (parent == _root)
{
_root = cur;
cur->_parent = nullptr;
}
else
{
if (ppnode->_left == parent)
{
ppnode->_left = cur;
}
else if (ppnode->_right == parent)
{
ppnode->_right = cur;
}
cur->_parent = ppnode;
}
cur->_bf = parent->_bf = 0;
}
//先右旋再左旋
void RotateRL(Node*parent)
{
//更新平衡因子
//分三种情况,因为curleft有可能是0,也有可能是1或者-1
//新增的为自己时,bf为0,新增在左是-1...
Node* cur = parent->_right;
Node* curleft = cur->_left;
int bf = curleft->_bf;//提前存一下
RotateR(parent->_right);
RotateL(parent);
//更新bf
if (bf == 0)
{
curleft->_bf = 0;
parent->_bf = 0;
cur->_bf = 0;
}
else if (bf == 1)
{
curleft->_bf = 0;
parent->_bf = -1;
cur->_bf = 0;
}
else if (bf == -1)
{
curleft->_bf = 0;
parent->_bf = 0;
cur->_bf = 1;
}
else
{
assert(false);
}
}
//先左旋再右旋
void RotateLR(Node* parent)
{
//更新平衡因子
//分三种情况,因为curright有可能是0,也有可能是1或者-1
//新增的为自己时,bf为0,新增在左是-1...
Node* cur = parent->_right;
Node* curright = cur->_left;
int bf = curright->_bf;
RotateL(parent->_left);
RotateR(parent);
//更新bf
if (bf == 0)
{
curright->_bf = 0;
parent->_bf = 0;
cur->_bf = 0;
}
else if (bf == 1)
{
curright->_bf = 0;
parent->_bf = 0;
cur->_bf = -1;
}
else if (bf == -1)
{
curright->_bf = 0;
parent->_bf = 1;
cur->_bf = 0;
}
else
{
assert(false);
}
}
void Print()
{
print(_root);
}
void print(const Node*root)
{
if (root == nullptr)
{
return;
}
print(root->_left);
cout << root->_kv.first << " ";
print(root->_right);
}
private:
Node* _root = nullptr;
};
题目知识点:
1:
AVL树:一棵AVL树或者是空树,或者是具有以下性质的二叉搜索树
1. 它的左右子树都是AVL树
2. 左右子树高度之差(简称平衡因子)的绝对值不超过1(-1/0/1)
故:如果一棵二叉搜索树是高度平衡的,它就是AVL树。如果它有n个结点,其高度可保持在O(logN),搜索时间复杂度O(logN)
A:AVL树也是二叉搜索树
AVL树没有极端情况,其是为了防止二叉搜索树的极端情况二给出的
C:AVL查询的时间复杂度是O(log_2N)
D:平衡因子:左右子树高度之间,其绝对值如果不超过1,则认为树就是平衡的