【数据结构与算法】-【平衡二叉树(AVL树)】

一、平衡二叉树

1.二叉排序树弊端

数列{1,2,3,4,5,6}，创建一颗二叉排序树(BST)，存在的问题分析:

①左子树全部为空，从形式上看，更像一个单链表。
②插入速度没有影响。
③查询速度明显降低(因为需要依次比较), 不能发挥BST的优势，因为每次还需要比较左子树，其查询速度比单链表还慢。

2.平衡二叉树基本介绍

平衡二叉树也叫平衡二叉搜索树（Self-balancing binary search tree）又被称为AVL树， 可以保证查询效率较高。
特点：它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。
平衡二叉树的常用实现方法有红黑树、AVL、替罪羊树、Treap、伸展树等。

3.平衡二叉树单旋转(左旋转)

对节点A进行左旋转的步骤

①将A 节点的右节点的左节点 ，指向 A节点；
②将A节点的右节点，指向A节点的右节点的左节点 ；

AVL树高度求解

// 创建Node结点
class Node {
	int value;
	Node left;
	Node right;

	public Node(int value) {

		this.value = value;
	}

	// 返回左子树的高度
	public int leftHeight() {
		if (left == null) {
			return 0;
		}
		return left.height();
	}

	// 返回右子树的高度
	public int rightHeight() {
		if (right == null) {
			return 0;
		}
		return right.height();
	}

	// 返回 以该结点为根结点的树的高度
	public int height() {
		return Math.max(left == null ? 0 : left.height(), right == null ? 0 : right.height()) + 1;
	}
}

---

AVL树左旋转

//左旋转方法
private void leftRotate() {
	
	//创建新的结点，以当前根结点的值
	Node newNode = new Node(value);
	//把新的结点的左子树设置成当前结点的左子树
	newNode.left = left;
	//把新的结点的右子树设置成带你过去结点的右子树的左子树
	newNode.right = right.left;
	//把当前结点的值替换成右子结点的值
	value = right.value;
	//把当前结点的右子树设置成当前结点右子树的右子树
	right = right.right;
	//把当前结点的左子树(左子结点)设置成新的结点
	left = newNode;		
}

---

3.平衡二叉树单旋转(右旋转)

对节点A进行右旋转的步骤

①将A节点的左节点的右节点 ，指向A节点；
②将A节点的左节点，指向A节点的左节点的右节点 ；

AVL树右旋转

//右旋转
private void rightRotate() {
	Node newNode = new Node(value);
	newNode.right = right;
	newNode.left = left.right;
	value = left.value;
	left = left.left;
	right = newNode;
}

---

4.平衡二叉树双旋转

在某些情况下，单旋转不能完成平衡二叉树的转换。

在添加节点时进行判定

// 添加结点的方法
// 递归的形式添加结点，注意需要满足二叉排序树的要求
public void add(Node node) {
	if (node == null) {
		return;
	}

	// 判断传入的结点的值，和当前子树的根结点的值关系
	if (node.value < this.value) {
		// 如果当前结点左子结点为null
		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();
			//然后在对当前结点进行左旋转
			leftRotate(); //左旋转..
		} else {
			//直接进行左旋转即可
			leftRotate();
		}
		return ; //必须要!!!
	}
	
	//当添加完一个结点后，如果 (左子树的高度 - 右子树的高度) > 1, 右旋转
	if(leftHeight() - rightHeight() > 1) {
		//如果它的左子树的右子树高度大于它的左子树的高度
		if(left != null && left.rightHeight() > left.leftHeight()) {
			//先对当前结点的左结点(左子树)->左旋转
			left.leftRotate();
			//再对当前结点进行右旋转
			rightRotate();
		} else {
			//直接进行右旋转即可
			rightRotate();
		}
	}
}