二分搜索树又名有序二叉查找树,它有一个特点是左子树的节点值要小于父节点值,右子树的节点值要大于父节点值。基于这样的特点,我们在查找某个节点的时候,可以采取二分查找的思想快速找到这个节点,时间复杂度期望值是为O(log n),但是它有最坏的的情况下。
例如,输入数组[9,7,5,3,1],如果要满足二分搜索树的规则插入一个个节点,这样的二叉树会退化成一条线性表,待会查找元素的时候时间复杂度已达O(N)。
所以针对这种情况,我们引申出了平衡二分搜索树,它每个节点的左右子树高度差不会超过1,它能在O(log n)内完成插入、查找和删除操作。平衡二分搜索树种类比较多,AVL树是其中的一种,但是它是最早被发明的自平衡二分搜索树。
AVL树也会被称为高度平衡树,因为它比二分搜索树多了一个特点:任一节点的左右子树高度差最大为1。如果以后去参赛,可以根据数据的情况更改高度差最大值,甚至也可以制定高度差相差几倍。
计算高度和平衡因子
计算高度是从叶子节点开始的,起始高度默认为1。然后计算叶子节点的父节点高度,比较左右子树的高度,采取最大值再加1就是这个节点的高度了。依此类推,直到整个树的顶点。
计算平衡因子也跟计算高度从叶子节点开始的,然后依次往父节点计算。节点的平衡因子公式是它左子树的高度减去它右子树的高度,有时候也会相反,可负数。
带有平衡因子-1、0或1的节点被认为是平衡的,即期望平衡节点的平衡因子的绝对值不会大于高度差最大值的。带有平衡因子-2或2的节点被认为是不平衡的,意味着需要重新调整这个树。平衡因子的绝对值最大值不会超过高度差最大值+1,说明这个数的任一节点的平衡因子不会出现-3或3。
如果改定高度差最大值为2,那么平衡因子会出现-3或3了,同时这个节点也是不平衡的,需要旋转调整。带有平衡因子-2、-1、0、1或2则被认为是平衡的。
动画
Code
左旋转和右旋转
AVL树调整不平衡的节点分为左旋转和右旋转,却分四种情况:LL、RR、LR和RL。其中L是左旋转,R是右旋转。如何采取使用哪一种情况则看插入的节点在哪里。
如果插入的节点是再T2子树里面,T1、T2、T3和T4都代表一个子树。T2里面插入一个节点,这个子树的高度加1,再计算父节点的平衡因子。如果这个节点是平衡的,则更新这个节点的高度。然后再往上计算父节点的平衡因子,接着判断是否平衡,如果是平衡的则更新高度,直到是不平衡的则进行旋转操作。
看上面图中,节点9是不平衡的,需要进行旋转操作。那如何进行哪种情况操作呢?
看左右子树哪一个子树插入了一个节点,节点5的左子树高度加1,导致节点5的平衡因子由0变成1。为了让节点5的平衡因子可以由1变成0,则希望节点5的右子树可以高度加1,所以就向节点5的父节点9进行右旋转操作,重新调整平衡因子,节点5的平衡因子恢复为0。
如果是下面情况,则不能单纯的进行右旋转操作了。看下面途中,插入一个节点是在节点3右子树发生的,节点3的平衡因子由0变成-1,应该希望是节点3左子树的高度可以高点。所以对节点3进行左旋转操作。
对节点3进行左旋转操作之后,更新相应节点的高度和平衡因子。看下面图中,发现节点5的平衡因子由-1变成1了,为了让1变成0,则希望是节点5的右子树高度可以高一点,所以对节点9进行右旋转操作。
进行LR旋转的操作如下图,节点5的平衡因子已恢复为0,节点9由开始的不平衡已变成平衡。
动画
Code
删除节点
AVL树的删除操作和二分搜索树一样,也分待删除结点的右子树为空、左子树为空和左右子树都不为空的情况。
那如何更新高度和平衡因子,不平衡的节点又如何调整为平衡的呢?和插入节点一样。
插入节点是插入一个节点后从叶子节点计算高度,然后再到父节点根据左右子树的高度计算平衡因子,接着更新高度,再到上一个父节点,直到整个二叉树的顶点。
删除节点可以看作是包含插入节点的,因为删除一个节点后会从左右子树中拉上来一个节点,不会再从叶子节点从新计算高度了,而是从左右子树开始接着更新高度和计算平衡因子。
动画
Code
长按下图二维码关注公众号,「算法无遗策」持续更新算法