AVL树总结

AVL树

概念

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

AVL树节点的定义

template<class T>
struct AVLTreeNode
{
 AVLTreeNode(const T& data)
     : _pLeft(nullptr), _pRight(nullptr), _pParent(nullptr)
 , _data(data), _bf(0)
 {}
 AVLTreeNode<T>* _pLeft;   // 左子
 AVLTreeNode<T>* _pRight;  // 右子
 AVLTreeNode<T>* _pParent; 
 T _data;
 int _bf;                  // 该节点的平衡因子
};

AVL树的插入

1. 按照二叉搜索树的方式插入新节点
2. 调整节点的平衡因子

AVL树的旋转

当插入一个新节点时，可能会造成祖先节点的平衡因子变为±2，此时就需要调整树的结构
新节点在较高左子树的左侧—左左：右单旋

/*
  上图在插入前，AVL树是平衡的，新节点插入到30的左子树中，30左子树增加一层，导致以60为根的二叉树不平衡，要让60平衡，只能将60左子树的高度减少一层，右子树增加一层，
  即将左子树往上提，这样60转下来，因为60比30大，只能将其放在30的右子树，而如果30有
右子树，右子树根的值一定大于30，小于60，只能将其放在60的左子树，旋转完成后，更新节点
的平衡因子即可。在旋转过程中，有以下几种情况需要考虑：
  1. 30节点的右孩子可能存在，也可能不存在
  2. 60可能是根节点，也可能是子树
     如果是根节点，旋转完成后，要更新根节点
     如果是子树，可能是某个节点的左子树，也可能是右子树
     a是小于30的一堆  b是大于30小于60的一堆 c是大于60的一堆
*/
void _RotateR(PNode 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(Node* parent)
	{
		Node* subL = parent->_left;
		Node* subLR = subL->_right;
		int bf = subLR->_bf;

		_RotateL(parent->_left);
		_RotateR(parent);

		subLR->_bf = 0;
		if (bf == 1)
		{
			parent->_bf = 0;
			subL->_bf = -1;
		}
		else if (bf == -1)
		{
			parent->_bf = 1;
			subL->_bf = 0;
		}
		else if (bf == 0)
		{
			parent->_bf = 0;
			subL->_bf = 0;
		}
		else
		{
			assert(false);
		}
	}

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

同理先进行右子树的右旋,在进行整体的左旋。
总结：
假如以pParent为根的子树不平衡，即pParent的平衡因子为2或者-2，分以下情况考虑

1. pParent的平衡因子为2，说明pParent的右子树高，设pParent的右子树的根为pSubR
   当pSubR的平衡因子为1时，执行左单旋
   当pSubR的平衡因子为-1时，执行右左双旋
2. pParent的平衡因子为-2，说明pParent的左子树高，设pParent的左子树的根为pSubL
   当pSubL的平衡因子为-1是，执行右单旋
   当pSubL的平衡因子为1时，执行左右双旋
   旋转完成后，原pParent为根的子树个高度降低，已经平衡，不需要再向上更新。