平衡二叉树

给你一个数列{1,2,3,4,5,6}，要求创建一颗二叉排序树(BST), 并分析问题所在.

上图BST 存在的问题分析:

1. 左子树全部为空，从形式上看，更像一个单链表.
2. 插入速度没有影响
3. 查询速度明显降低(因为需要依次比较), 不能发挥BST 的优势，因为每次还需要比较左子树，其查询速度比 单链表还慢
4. 解决方案-平衡二叉树(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}