平衡搜索树——AVL树

AVL树的发展

二叉搜索树虽可以缩短查找的效率，但如果数据有序或接近有序二叉搜索树将退化为单支树，查找元素相当于在顺序表中搜索元素，效率低下。因此，两位俄罗斯的数学家G.M.Adelson-Velskii和E.M.Landis在1962年发明了一种解决上述问题的方法：当向二叉搜索树中插入新结点后，如果能保证每个结点的左右子树高度之差的绝对值不超过1(需要对树中的结点进行调整)，即可降低树的高度，从而减少平均搜索长度。
例如：
采用平衡因子，平衡因子=右子树的高度-左 子树的高度
采用平衡因子对数树进行调整得到的后的树：

AVL树节点的定义

AVL树节点的定义：

template<class K, class V>
class AVLTreeNode
{
	AVLTreeNode<K, V>*_left;
	AVLTreeNode<K, V>*_right;
	AVLTreeNode<K, V>*_parent;
	int _bf;//平衡因子
	pair<K, V>_kv;//组成一个pair结构就可以返回多个值
};

AVL树的插入

树的插入可以分为两步：
1.按照二叉搜索树的方式插入新节点
2.调整节点的平衡因子

bool Insert(const T& data) {
// 1. 先按照二叉搜索树的规则将节点插入到AVL树中
 // ...
 
 // 2. 新节点插入后，AVL树的平衡性可能会遭到破坏，此时就需要更新平衡因子，并检测是否破坏了
AVL树
 // 的平衡性
 
 /*
 pCur插入后，pParent的平衡因子一定需要调整，在插入之前，pParent
 的平衡因子分为三种情况：-1，0, 1, 分以下两种情况：
 1. 如果pCur插入到pParent的左侧，只需给pParent的平衡因子-1即可
 2. 如果pCur插入到pParent的右侧，只需给pParent的平衡因子+1即可
 
 此时：pParent的平衡因子可能有三种情况：0，正负1， 正负2
 1. 如果pParent的平衡因子为0，说明插入之前pParent的平衡因子为正负1，插入后被调整成0，此
时满足
 AVL树的性质，插入成功
 2. 如果pParent的平衡因子为正负1，说明插入前pParent的平衡因子一定为0，插入后被更新成正负
1，此
 时以pParent为根的树的高度增加，需要继续向上更新
 3. 如果pParent的平衡因子为正负2，则pParent的平衡因子违反平衡树的性质，需要对其进行旋转
处理
 */
	 while (pParent)
	 {
		 // 更新双亲的平衡因子
		 if (pCur == pParent->_pLeft)
		 pParent->_bf--;
		 else
		 pParent->_bf++;
		 // 更新后检测双亲的平衡因子
		 if (0 == pParent->_bf)
		 break;
		 else if (1 == pParent->_bf || -1 == pParent->_bf)
		 {
		 // 插入前双亲的平衡因子是0，插入后双亲的平衡因为为1 或者 -1 ，说明以双亲为根的
		二叉树
		 // 的高度增加了一层，因此需要继续向上调整
		 pCur = pParent;
		 pParent = pCur->_pParent;
		 }
		 else
		 {
		 // 双亲的平衡因子为正负2，违反了AVL树的平衡性，需要对以pParent
		 // 为根的树进行旋转处理
		 if(2 == pParent->_bf)
		 {
		 // ...
		 }
		 else
		 {
		 // ...
		 }
		 }
		 }
	 return true; 
}

AVL树的旋转

1.新节点插入较高左子树的左侧——左左：右单旋

void _RotateR(PNode pParent)
 {
	 // pSubL: pParent的左孩子
	 // pSubLR: pParent左孩子的右孩子，注意：该
	 PNode pSubL = pParent->_pLeft;
	 PNode pSubLR = pSubL->_pRight;
	 // 旋转完成之后，30的右孩子作为双亲的左孩子
	 pParent->_pLeft = pSubLR;
	 // 如果30的左孩子的右孩子存在，更新亲双亲
	 if(pSubLR)
	 pSubLR->_pParent = pParent;
	 // 60 作为 30的右孩子
	 pSubL->_pRight = pParent;
	 // 因为60可能是棵子树，因此在更新其双亲前必须先保存60的双亲
	 PNode pPParent = pParent->_pParent;
	 
	 // 更新60的双亲
	 pParent->_pParent = pSubL;
	 
	 // 更新30的双亲
	 pSubL->_pParent = pPParent;
	 // 如果60是根节点，根新指向根节点的指针
	 if(NULL == pPParent)
	 {
	 _pRoot = pSubL;
	 pSubL->_pParent = NULL;
	 }
	 else
	 {
	 // 如果60是子树，可能是其双亲的左子树，也可能是右子树
	 if(pPParent->_pLeft == pParent)
	 pPParent->_pLeft = pSubL;
	 else
	 pPParent->_pRight = pSubL;
	 }
	 // 根据调整后的结构更新部分节点的平衡因子
	 pParent->_bf = pSubL->_bf = 0; 
}

2.新节点插入较高右子树的右侧——右右：左单旋

3.新节点插入较高左子树的右侧——左右：先左旋再右单旋

// 旋转之前，60的平衡因子可能是-1/0/1，旋转完成之后，根据情况对其他节点的平衡因子进行调整
void _RotateLR(PNode pParent) 
{
	 PNode pSubL = pParent->_pLeft;
	 PNode pSubLR = pSubL->_pRight;
	 
	 // 旋转之前，保存pSubLR的平衡因子，旋转完成之后，需要根据该平衡因子来调整其他节点的平
	衡因子
	 int bf = pSubLR->_bf;
	 
	 // 先对30进行左单旋
	 _RotateL(pParent->_pLeft);
	 
	 // 再对90进行右单旋
	 _RotateR(pParent);
	 if(1 == bf)
	 pSubL->_bf = -1;
	 else if(-1 == bf)
	 pParent->_bf = 1; 
 }

新节点插入在较高右子树的左侧——右左：先右单旋再左单旋