第二十六题：判断一颗二叉树是不是平衡二叉树（Java）

题目要求：

如标题：

什么是平衡二叉树？任意节点的左子树和右子树的高度差不超过1，空数也是一颗平衡二叉树。

要求：二叉树的节点的个数是N，时间复杂度是O（N）

代码实现与分析：

一个树是平衡二叉树的条件是，每一个子树都是平衡二叉树。对于node这颗子树，在何时是一颗平衡二叉树？

①：左子树不平衡，false

②：右子树不平衡，false

③：左子树平衡，拿到左子树的高度h1

④：右子树平衡，拿到右子树的高度h2，然后看两者的高度差是否大于1。

有了以上的信息，我们就能判断，node节点是否是一颗平衡二叉树，第一步，我们分析了所有的可能性，整理出信息，接下来第二步：设计我们的递归返回结构，因为我们整体是在递归函数中做一个判断，所以左树应给给一个返回值，这个返回值里面包含，左树是否平衡信息和左树的高度信息，右树也如此，node这棵树的返回值也是如此。 

package com.isea.brush.tree;

/**
 * 判断一棵树是否是平衡二叉树
 */
public class IsBalanceTree {
    private static class Node{
        private Integer value;
        private Node left;
        private Node right;

        public Node(int value){
            this.value = value;
        }
    }
    private static class ReturnData{
        private boolean isB;
        private int h;
        public ReturnData(boolean isB,int h){
            this.h = h;
            this.isB = isB;
        }
    }

    public static boolean isB(Node head){
        return process(head).isB;
    }

    /**
     * 列出所有的可能性之后，确定我们需要的返回值信息，然后设计我们的递归函数,要收集什么样子的信息，按照
     * 我们的子过程(左右子树)交给我的信息，自己也加工为这样的信息，往上级结构返回
     * @param head
     * @return
     */
    public static ReturnData process(Node head){
       if (head == null){
           return new ReturnData(true,0);
       }
       ReturnData leftData = process(head.left);
       if (!leftData.isB){
           return new ReturnData(false,0);
       }
       ReturnData rightData = process(head.right);
       if (!rightData.isB){
           return new ReturnData(false,0);
       }
       if (Math.abs(leftData.h - rightData.h) > 1){
           return new ReturnData(false,0);
       }
       return new ReturnData(true,Math.max(leftData.h,rightData.h) + 1);
    }

    public static void main(String[] args) {
        Node head = new Node(1);
        head.left = new Node(2);
        head.right = new Node(3);
        head.left.left = new Node(4);
        head.left.right = new Node(5);
        head.right.left = new Node(6);
        head.right.right = new Node(7);

        System.out.println(isB(head)); // true

    }
}