Java——平衡二叉树

一、基本概念：

平衡二叉树是对 二叉排序树的改进。

二、实现：

首先要根据条件判断是否要进行平衡二叉树的调整：要获取到左右子树高度。

  //********************************************************
    //计算高度
    //返回当前节点的高度，（以该节点为根节点的树的高度）
    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 left.height();
    }
    //返回右子树的高度
    public int rightHeight(){
        if (right == null){
            return 0;
        }
        return right.height();
    }

1、单旋转：

（1）左旋转：

public void leftRotate(){
        //以当前节点创建新的节点
        Node newNode = new Node(value);
        //将当前节点的左子树设置成新的节点的左子树
        newNode.left = this.left;
        //将当前节点的右子树的左子树设置为新结点的右子树
        newNode.right = this.right.left;
        //将当前节点的值替换为右子节点的值
        this.value = this.right.value;
        //将当前节点的右子树的右子树设置成当前节点的右子树
        this.right = this.right.right;
        //将新结点设置为当前的节点的左子树
        this.left = newNode;
    }

（2）右旋转：

 //右旋转
    public void rightRotate(){
        //以当前节点创建新的节点
        Node newNode = new Node(value);
        //将当前节点的右子树设置成新的节点的右子树
        newNode.right = this.right;
        //将当前节点的左子树的右子树设置为新结点的左子树
        newNode.left = this.left.right;
        //将当前节点的值替换为左子节点的值
        this.value = this.left.value;
        //将当前节点的左子树的左子树设置成当前节点的左子树
        this.left = this.left.left;
        //将新结点设置为当前的节点的右子树
        this.right = newNode;
    }

2、双旋转：

前面的两个数列，进行单旋转(即一次旋转)就可以将非平衡二叉树转成平衡二叉树,但是在某些情况下，单旋转 不能完成平衡二叉树的转换。
思路分析：当首先符合 右旋转的条件下：
（1）如果当前节点的左子树的 （右子树高度 大于 左子树的高度）
（2）就先对这个左子树进行 左旋转
（3）然后对当前节点进行右旋转
如果首先符合右子树，则与此相反。

将平衡的过程代码添加在 节点的添加中，添加完了之后就进行判断，看是否需要进行平衡。

 //添加节点，需要满足二叉排序树的要求（以递归的方式进行）
    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);
            }
        }

        //当添加完一个节点后，判断是否需要进行旋转平衡
        if (rightHeight() - leftHeight() > 1){//右边高，进行左旋转
            if (right != null && right.leftHeight() > right.rightHeight()){
                right.rightRotate();
                leftRotate();
            }else {
                leftRotate();
            }
            return;
        }

        if (leftHeight() - rightHeight() > 1){// 左边高，进行右旋转
            //如果左子树作为根节点下的 （右子树高度 大于 左子树的高度）
            if (left != null && left.rightHeight() > left.leftHeight()){
                //就先对这个左子树进行 左旋转
                left.leftRotate();
                //然后对当前节点进行右旋转
                rightRotate();
            }else {
                rightRotate();
            }
        }
    }