牛客笔试题---实现一个函数，检查二叉树是否平衡

题目描述:

实现一个函数，检查二叉树是否平衡，平衡的定义如下，对于树中的任意一个结点，其两颗子树的高度差不超过1。
给定指向树根结点的指针TreeNode* root，请返回一个bool，代表这棵树是否平衡

代码展示: 

public class Balance {
    class TreeNode{
        TreeNode root = null;
        TreeNode left = null;
        TreeNode right = null;
        int val;

        TreeNode(int x){
            x=this.val;
        }
    }
    public boolean isBalance(TreeNode root){
        //判断根是否为空
        if(root == null){
            return true;//空树即是一颗平衡二叉树
        }
        //获取左子树的高度
        int leftHeight = getTreeHeight(root.left);
        int rightHeight = getTreeHeight(root.right);

        //左边子树高度大于1表示不是平衡二叉树
        if(Math.abs(leftHeight - rightHeight) > 1){
            return false;
        }

        //检查是否平衡
        return isBalance(root.left) && isBalance(root.right);
    }

    private int getTreeHeight(TreeNode root) {
        if(root == null){
            return 0;
        }
        return Math.max(getTreeHeight(root.left),getTreeHeight(root.right))+1;

    }
}