普通的二叉查找树(BST)虽然已经实现了对于节点的快速查找,但是如果树的拓扑结构没有设计正确,例如讲一个有序序列存入BST中,就会使BST的二分查找能力损失,也就是常说的失去了平衡。为了保证BST的查找能力,在BST形成过程中进行平衡调整,就形成了平衡二叉查找树,简称平衡二叉树(AVL-tree)。

AVL具备以下性质:

  1. 从任何一个节点出发,左子树和右子树的深度差的绝对值不超过1。这个深度差被称作平衡因子
  2. 任何一个节点的左子树和右子树都是平衡二叉树。

通过AVL所实现的拓扑结构,就保证了树的二分查找能力,AVL在查找、插入和删除时的复杂度均为平衡二叉树 - 图1

AVL构建

要构建一棵AVL树,我们依旧可以使用前面定义的节点类,但是为了能够适应AVL树的一些额外属性和操作,我们还是需要将这个节点类扩展一下。

  1. public class AVLNode extends BSTNode {
  2. private Integer depth;
  3. public AVLNode(Integer value) {
  4. super(value);
  5. this.depth = 0;
  6. }
  7. public Integer getDepth() {
  8. return this.depth;
  9. }
  10. public Integer setDepth(Integer depth) {
  11. this.depth = depth;
  12. }
  13. public void increaseDepth() {
  14. this.depth++;
  15. }
  16. public void decreaseDepth() {
  17. this.depth--;
  18. }
  19. }

实例中的AVL树节点类增加了一个用于保存节点深度的属性,这个深度属性可以用来比较得出是否平衡的结论。

树的旋转

向一个AVL中插入或者删除一个节点,会导致AVL的不平衡。例如以下示例。
AVL插入节点.svg
左侧的AVL在插入一个新的节点5之后,就变得不平衡了。节点50的左子树的高度为4,右子树的高度为2,AVL已经失去了平衡。在这种失衡的情况下,AVL是通过旋转最小失衡子树来重新获取平衡的。那么这里就引入了一个新的名词:最小失衡子树

最小失衡子树是指从新插入的节点向根查找,第一个平衡因子的绝对值超过1的节点即为最小失衡子树的根节点。一棵失衡的AVL树中,是可能会同时存在多棵失衡子树的,但是要使AVL树恢复平衡,只需要调整最小的失衡子树即可。

对失衡子树的调整是通过旋转来完成的,旋转子树的目的是降低树的高度。子树的旋转有两个方向:左旋和右旋,使用哪个方向是由子树的高度决定的。如果节点的右子树高度比较高,那么就采用左旋,降低右子树高度;反之,使用右旋可以降低左子树高度。

左旋

最小失衡子树的左旋可以遵循以下步骤处理:

  1. 子树根节点(ori)的右孩子(rc)替代根节点成为新的根节点。
  2. 右孩子(rc)的左子树变为原根节点(ori)的右子树。
  3. 原根节点(ori)变为右孩子(rc)的左子树。

具体操作实现可以参考以下例程。

  1. public class AVLTree {
  2. public void rotateLeft(AVLNode root) {
  3. root.getRightChild().ifPresent(rightChild -> {
  4. root.getParent().ifPresent(parent -> {
  5. if (root.isLeftChild()) {
  6. parent.attachLeftChild(rightChild);
  7. } else {
  8. parent.attachRightChild(rightChild);
  9. }
  10. });
  11. rightChild.getLeftChild().ifPresent(root::attachRightChild);
  12. rightChild.attachLeftChild(root);
  13. });
  14. }
  15. }

右旋

最小失衡子树的右旋非左旋正好相反,可以遵循以下步骤:

  1. 子树根节点(ori)的左孩子(lc)替代根节点成为新的根节点。
  2. 左孩子(lc)的右子树变为原根节点(ori)的左子树。
  3. 原根节点(ori)变为左孩子(lc)的右子树。

具体操作实现可以参考以下例程。

  1. public class AVLTree {
  2. public void rotateRight(AVLNode root) {
  3. root.getLeftChild().ifPresent(leftChild -> {
  4. root.getParent().ifPresent(parent -> {
  5. if (root.isLeftChild()) {
  6. parent.attachLeftChild(leftChild);
  7. } else {
  8. parent.attachRightChild(leftChild);
  9. }
  10. });
  11. leftChild.getRightChild().ifPresent(root::attachLeftChild);
  12. leftChild.attachRightChild(root);
  13. });
  14. }
  15. }

节点插入

向AVL中的某一个节点k的左右子树上插入一个新节点n,会有以下四种情况可以破坏原有AVL的平衡性:

  1. 在节点k的左子树根节点的左子树上插入节点n(简称LL插入)。要重新达到平衡需要执行右旋操作。
  2. 在节点k的右子树根节点的右子树上插入节点n(简称RR插入)。要重新达到平衡需要执行左旋操作。
  3. 在节点k的左子树根节点的右子树上插入节点n(简称LR插入)。要重新达到平衡需要先执行左旋操作,再执行右旋操作。
  4. 在节点k的右子树根节点的左子树上插入节点n(简称RL插入)。要重新达到平衡需要先执行右旋操作,再执行左旋操作。

由于新节点都是作为叶子节点插入,所以在完成节点插入以后,需要完成一项工作,就是更新所有父代节点的深度值。这个更新不需要遍历整棵树,只需要遍历新节点的所有父代路径即可。以下是深度值更新的示例。

  1. public class AVLTree {
  2. public void updateDepth(AVLNode node) {
  3. if (node.isLeaf()) {
  4. node.setDepth(0);
  5. } else {
  6. Integer leftChildDepth = node.getLeftChild().get(AVLNode::getDepth).orElse(0);
  7. Integer rightChildDepth = node.getRightChild().get(AVLNode::getDepth).orElse(0);
  8. node.setDepth(IntStream
  9. .of(leftChildDepth, rightChildDepth)
  10. .map(d -> d + 1)
  11. .max()
  12. .orElse(0));
  13. }
  14. node.getParent().ifPresent(this::upateDepth);
  15. }
  16. }

虽然在节点插入后的重新平衡有四种情况,但是总结起来,只是一个不断遍历新插入节点的父级路径,使其父级路径重新平衡的过程。而在这个过程中,针对每一个节点,可以判断其左右子树的深度值,如果左侧的深度大于右侧的深度,那么就采用右旋进行平衡;反之采用左旋进行平衡。

以下是一个处理新插入节点并实现再平衡的示例。

  1. public class AVLTree {
  2. public void insert(Integer value) {
  3. // 这个pointer十分有用,可以用来对树进行遍历和操作
  4. AVLNode pointer = this.root;
  5. ANLNode newNode = new AVLNode(value);
  6. // 首先完成节点的插入
  7. while (!pointer.isLeaf()) {
  8. if (pointer.compareTo(newNode) > 0) {
  9. pointer = pointer.getRightChild().get();
  10. continue;
  11. }
  12. if (pointer.compareTo(newNode) < 0) {
  13. pointer = pointer.getLeftChild().get();
  14. continue;
  15. }
  16. if (pointer.compareTo(newNode) == 0) {
  17. throw new RepeatValueException();
  18. }
  19. }
  20. if (newNode.compareTo(pointer) < 0) {
  21. pointer.attachLeftChild(newNode);
  22. } else {
  23. pointer.attachRightChild(newNode);
  24. }
  25. this.updateDepth(newNode);
  26. // 然后开始对新节点的父级路径进行再平衡
  27. pointer = newNode;
  28. while (pointer.getParent().isPresent()) {
  29. Integer leftChildDepth = pointer.getLeftChild().map(AVLNode::getDepth).orElse(0);
  30. Integer rightChildDepth = pointer.getRightChild().map(AVLNode::getDepth).orElse(0);
  31. if (Math.abs(leftChildDepth - rightChildDepth) > 1) {
  32. if (leftChildDepth > rightChildDepth) {
  33. this.rightRotate(pointer);
  34. } else {
  35. this.leftRotate(pointer);
  36. }
  37. this.updateDepth(pointer);
  38. pointer = newNode;
  39. } else {
  40. pointer = pointer.getParent().get();
  41. }
  42. }
  43. }
  44. }

实例代码中虽然不仅对仅需要再平衡的最小失衡子树进行了平衡,而且也同时扫描了新插入节点的所有父级路径。这样做的目的主要是把需要两次旋转的操作化简到了一个循环中。由于每次仅会扫描一条路径,所以增加的复杂度有限。

节点删除

AVL中的节点删除与BST中的节点删除操作是相同的,只是AVL在完成删除操作以后需要修正所有的不平衡节点。一般都会分为以下四种情况来处理。

  1. 被删除节点是叶子节点。
  2. 被删除节点只有左子树。
  3. 被删除节点只有右子树。
  4. 被删除节点既有左子树又有右子树。

在完成节点的删除以后,可以选择被删除节点的左右子树中深度较大的那一支,来更新整条路径上的节点深度,并进行再平衡操作。

AVL的缺点

AVL在大量查找操作的情况下效率会更高,但是对于增删操作,AVL会进行大量的再平衡操作,这样就大大降低了AVL的性能。所以在查找操作远大于增删操作次数的时候,使用AVL会得到更高的效率。如果在树中的增删操作比较多,那么可以选择增删性能更高的红黑树。