给你一个数列{1,2,3,4,5,6},要求创建一颗二叉排序树(BST), 并分析问题所在.
上图BST 存在的问题分析:
- 左子树全部为空,从形式上看,更像一个单链表.
- 插入速度没有影响
- 查询速度明显降低(因为需要依次比较), 不能发挥BST 的优势,因为每次还需要比较左子树,其查询速度比 单链表还慢
- 解决方案-平衡二叉树(AVL)
平衡二叉树
平衡二叉树也叫平衡二叉搜索树(Self-balancing binary search tree)又被称为AVL树, 可以保证查询效率较高。
具有以下特点:它是一 棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。平衡二叉树的常用实现方法有红黑树、AVL、替罪羊树、Treap、伸展树等。
左旋转
给你一个数列,创建出对应的平衡二叉树.数列 {4,3,6,5,7,8}
public void leftRotate(){
Node newNode=new Node(id);
newNode.left=left;
newNode.right=right.left;
id=right.id;
right=right.right;
left=newNode;
}
右旋转
public void rightRotate(){
Node newNode=new Node(id);
newNode.left=left.right;
newNode.right=right;
id=left.id;
left=left.left;
right=newNode;
}
双旋转
前面的两个数列,进行单旋转(即一次旋转)就可以将非平衡二叉树转成平衡二叉树,但是在某些情况下,单旋转不能完成平衡二叉树的转换。比如数列 int[] arr = { 10, 11, 7, 6, 8, 9 }; 运行原来的代码可以看到,并没有转成 AVL树. int[] arr = {2,1,6,5,7,3}; // 运行原来的代码可以看到,并没有转成 AVL树
问题分析
1.当符号右旋转的条件时
2.如果它的左子树的右子树高度大于它的左子树的高度
3.先对当前这个结点的左节点进行左旋转
4.在对当前结点进行右旋转的操作即可
//当添加完一个结点后,如果: (右子树的高度-左子树的高度) > 1 , 左旋转
if(rightHeight()-leftHeight()>1){
//如果它的右子树的左子树的高度大于它的右子树的右子树的高度
if(right!=null&&right.leftHeight()>left.rightHeight()){
//先对右子结点进行右旋转
right.rightRotate();
//然后在对当前结点进行左旋转
leftRotate();//左旋转..
}else{
//直接进行左旋转即可
leftRotate();
}
return;
}
//当添加完一个结点后,如果 (左子树的高度 - 右子树的高度) > 1, 右旋转
if(leftHeight()-rightHeight()>1){
//如果它的左子树的右子树高度大于它的左子树的高度
if(left!=null&&left.leftHeight()<left.rightHeight()){
//先对当前结点的左结点(左子树)->左旋转
left.leftRotate();
//再对当前结点进行右旋转
rightRotate();
}else{
//直接进行右旋转即可
rightRotate();
}
}
完整代码如下:
package com.zx.ds.avl;
class Node {
int id;
Node left;
Node right;
public Node(int id) {
this.id = id;
}
@Override
public String toString() {
return "Node{" +
"id=" + id +
'}';
}
public int height(){
return Math.max(left==null?0:left.height(),right==null?0:right.height())+1;
}
public int leftHeight(){
if(left==null){
return 0;
}
return this.left.height();
}
public int rightHeight(){
if(right==null){
return 0;
}
return this.right.height();
}
//添加结点的方法
//递归的形式添加结点,注意需要满足二叉排序树的要求
public void add(Node node) {
if (node == null) {
return;
}
if (this.id > node.id) {
if (this.left == null) {
this.left = node;
} else {
this.left.add(node);
}
}
if (this.id < node.id) {
if (this.right == null) {
this.right = node;
} else {
this.right.add(node);
}
}
//当添加完一个结点后,如果: (右子树的高度-左子树的高度) > 1 , 左旋转
if(rightHeight()-leftHeight()>1){
//如果它的右子树的左子树的高度大于它的右子树的右子树的高度
if(right!=null&&right.leftHeight()>left.rightHeight()){
//先对右子结点进行右旋转
right.rightRotate();
//然后在对当前结点进行左旋转
leftRotate();//左旋转..
}else{
//直接进行左旋转即可
leftRotate();
}
return;
}
//当添加完一个结点后,如果 (左子树的高度 - 右子树的高度) > 1, 右旋转
if(leftHeight()-rightHeight()>1){
//如果它的左子树的右子树高度大于它的左子树的高度
if(left!=null&&left.leftHeight()<left.rightHeight()){
//先对当前结点的左结点(左子树)->左旋转
left.leftRotate();
//再对当前结点进行右旋转
rightRotate();
}else{
//直接进行右旋转即可
rightRotate();
}
}
}
public void leftRotate(){
Node newNode=new Node(id);
newNode.left=left;
newNode.right=right.left;
id=right.id;
right=right.right;
left=newNode;
}
public void rightRotate(){
Node newNode=new Node(id);
newNode.left=left.right;
newNode.right=right;
id=left.id;
left=left.left;
right=newNode;
}
//中序遍历
public void midOrder() {
if (this == null) {
System.out.println("tree is null");
return;
}
if (this.left != null) {
this.left.midOrder();
}
System.out.println(this);
if (this.right != null) {
this.right.midOrder();
}
}
//查找要删除的结点
public Node search(int no) {
if (this.id == no) {
return this;
}
if (this.id > no && this.left != null) {
return this.left.search(no);
}
if (this.id < no && this.right != null) {
return this.right.search(no);
}
return null;
}
//查找要删除结点的父结点
public Node searchParent(int no) {
if ((this.left != null && this.left.id == no) || (this.right != null && this.right.id == no)) {
return this;
}
if (this.id > no && this.left != null) {
return this.left.searchParent(no);
}
if (this.id < no && this.right != null) {
return this.right.searchParent(no);
}
return null;
}
}
class AVLTree{
private Node root;
public Node getRoot() {
return root;
}
//添加结点的方法
public void add(Node node) {
if (root == null) {
//如果root为空则直接让root指向node
root = node;
} else {
this.root.add(node);
}
}
//中序遍历
public void midOrder() {
if (this.root == null) {
System.out.println("tree is null");
return;
} else {
this.root.midOrder();
}
}
//查找要删除的结点
public Node search(int no) {
if (this.root == null) {
return null;
} else {
return this.root.search(no);
}
}
//查找父结点
public Node searchParent(int no) {
if (this.root == null) {
return null;
} else {
return this.root.searchParent(no);
}
}
//编写方法:
//1. 返回的 以node 为根结点的二叉排序树的最小结点的值
//2. 删除node 为根结点的二叉排序树的最小结点
/**
*
* @param node 传入的结点(当做二叉排序树的根结点)
* @return 返回的 以node 为根结点的二叉排序树的最小结点的值
*/
public int deleteRightTreeMin(Node node){
Node target=node;
//循环的查找左子节点,就会找到最小值
while(target.left!=null){
target=target.left;
}
//这时 target就指向了最小结点
//删除最小结点
deleteNode(target.id);
return target.id;
}
//删除结点
public void deleteNode(int no) {
if (this.root == null) {
return;
} else {
//1.需求先去找到要删除的结点 targetNode
Node targetNode = search(no);
//如果没有找到要删除的结点
if (targetNode == null) {
System.out.println("this node is not exist");
return;
}
//如果我们发现当前这颗二叉排序树只有一个结点
if (root.left == null && root.right == null) {
root = null;
return;
}
//去找到targetNode的父结点
Node parent = searchParent(no);
//如果要删除的结点是叶子结点
if (targetNode.left == null && targetNode.right == null) {
//判断targetNode 是父结点的左子结点,还是右子结点
if (parent.left == targetNode) {
parent.left = null;
}
if (parent.right == targetNode) {
parent.right = null;
}
} else if (targetNode.left != null && targetNode.right != null) {//删除有两颗子树的节点
int temp=deleteRightTreeMin(targetNode);
targetNode.id=temp;
} else {// 删除只有一颗子树的结点
//如果要删除的结点有左子结点
if (targetNode.left != null) {
if (parent != null) {
//如果 targetNode 是 parent 的左子结点
if (parent.left == targetNode) {
parent.left = targetNode.left;
} else {
parent.right = targetNode.left;
}
} else {
root = targetNode.left;
}
} else {
if (parent != null) {
if (parent.left == targetNode) {
parent.left = targetNode.right;
} else {
parent.right = targetNode.right;
}
} else {
root = targetNode.right;
}
}
}
}
}
}
public class AVLTreeDemo {
public static void main(String[] args) {
//int array[]={1,2,3,4,5,6};
//int array[]={4,3,6,5,7,8};
//int array[]={10,12,8,9,7,6};
int array[]={10,11,7,6,8,9};
AVLTree tree=new AVLTree();
for(int i=0;i<array.length;i++){
tree.add(new Node(array[i]));
}
System.out.println("中序遍历:");
tree.midOrder();
System.out.println("平衡处理后:");
System.out.println("树的高度:"+tree.getRoot().height());
System.out.println("树的左子树高度:"+tree.getRoot().leftHeight());
System.out.println("树的右子树高度:"+tree.getRoot().rightHeight());
System.out.println("根节点"+tree.getRoot());
}
}
中序遍历:
Node{id=6}
Node{id=7}
Node{id=8}
Node{id=9}
Node{id=10}
Node{id=11}
平衡处理后:
树的高度:3
树的左子树高度:2
树的右子树高度:2
根节点Node{id=8}