一、基本概念
- 平衡二叉树也称平衡二叉搜索树,又被叫做AVL树,可以保证查询效率较高。
- 平衡二叉树的特点:它是一棵空树或者它的左右两个子树高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。
- 图例:
二、思路分析
1. 平衡二叉树的构建:因为平衡二叉树本质也是一棵二叉排序树,所以我们可以使用旋转的方式来对平衡二叉树进行构建。
- 当右子树高度-左子树高度>1时,我们利用左旋转来增加左子树的高度。
如同所示的一棵二叉排序树,很明显,这颗二叉排序树并非平衡二叉树。我们可以通过以下几步进行左旋转得到一棵平衡二叉树:- 创建一个新节点newNode,并赋值为当前节点(根节点)的值。
- 将当前节点的左子树设置为新节点newNode的左子树
- 将当前节点的右子树的左子树设置为新节点newNode的右子树
- 将当前节点的值设置为右子树的值
- 将当前节点的右子树设置为右子树的右子树
- 将当前节点的左子树设置为新节点newNode
- 创建一个新节点newNode,并赋值为当前节点(根节点)的值。
- 类似的,当左子树高度-右子树高度>1时,我们利用右旋转来增加右子树的高度。
- 在我们进行旋转时,可能进行一次旋转后还是无法得到一棵平衡二叉树,这时我们需要进行双旋转:
- 当前节点符合右(左)旋转的条件
- 它的左(右)子树的右(左)子树高度大于它的左(右)子树高度
- 先对当前节点的左(右)子树进行左(右)旋转
- 再对当前节点进行右旋转
三、代码实现
1.左旋转方法
//左旋转
public void letfRotate() {
//创建一个新节点,值等于当前根节点的值
Node newNode = new Node(value);
//把新节点的左子树设置成当前节点的左子树
newNode.left = left;
//把新节点的右子树设置成当前节点的右子树的左子树
newNode.right = right.left;
//把当前节点的值换为右子节点的值
value = right.value;
//把当前节点的右子树设置为右子树的右子树
right = right.right;
//把当前节点的左子树设置为新节点
left = newNode;
}
2.右旋转方法
//右旋转
public void rightRotate() {
Node newNode = new Node(value);
newNode.right = right;
newNode.left = left.right;
value = left.value;
left = left.left;
right = newNode;
}
3.利用双旋转添加节点构建平衡二叉树
//返回左子树的高度
public int leftHeight() {
if (left == null) {
return 0;
} else {
return left.height();
}
}
//返回右子树的高度
public int rightHeight() {
if (right == null) {
return 0;
} else {
return right.height();
}
}
//返回当前节点的高度,以该节点为根节点的树的高度
public int height() {
return Math.max(left == null ? 0 : left.height(), right == null ? 0 : right.height())+1;
}
//递归添加节点
public void add(Node node) {
if (node == null) {
return;
}
if (node.value < this.value) {
if (this.left == null) {
this.left = node;
} else {
this.left.add(node);
}
} else {
if (this.right == null) {
this.right = node;
} else {
this.right.add(node);
}
}
//当添加完一个节点后,如果右子树高度-左子树高度>1,则左旋转
if (rightHeight() - leftHeight() > 1) {
if (right != null && right.leftHeight() > right.rightHeight()) {
right.rightRotate();
letfRotate();
} else {
rightRotate();
}
return;
}
if (leftHeight() - rightHeight() > 1) {//如果左子树高度-右子树高度>1,则右旋转
if (left != null && left.rightHeight() > left.leftHeight()) {//如果它的左子树的右子树高度大于左子树的高度
//先对当前节点的左子树进行左旋转
left.letfRotate();
//再对当前节点进行右旋转
rightRotate();
} else {
rightRotate();
}
}
}
4.这样我们给定一个序列就可以构建一棵平衡二叉树
public class AVLTreeDemo {
public static void main(String[] args) {
// int[] arr = {4, 3, 6, 5, 7, 8};
int[] arr = {10,11,7,6,8,9};
AVLTree avlTree = new AVLTree();
for (int i = 0; i < arr.length; i++) {
avlTree.add(new Node(arr[i]));
}
System.out.println("中序遍历");
avlTree.midOrder();
System.out.println("树的高度:" + avlTree.getRoot().height());
System.out.println("左子树的高度:" + avlTree.getRoot().leftHeight());
System.out.println("右子树的高度:" + avlTree.getRoot().rightHeight());
System.out.println("根节点:" + avlTree.getRoot());
System.out.println("根节点的左子节点:" + avlTree.getRoot().left.left);
}
}
测试结果: