平衡二叉树从顶到底和从底到顶

1. 平衡二叉树

描述：

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。

1.1 从顶到底

思路：

从顶到底，O(N^2)，基础

（1）定义辅助方法depth求得当前树的高度
（2）若root为空，是平衡的，返回true
（3）求左子树和右子树的高度
（4）若左子树和右子树高度差的绝对值不超过1（利用Math.abs()求绝对值）并且左右子树都平衡（递归），返回true

代码：

class Solution {
    public boolean isBalanced(TreeNode root) {
        if(root == null) return true;
        int leftdepth = depth(root.left);
        int rightdepth = depth(root.right);

        return Math.abs(leftdepth - rightdepth) < 2 && isBalanced(root.left) && isBalanced(root.right);

    }
    public int depth(TreeNode root){
        if(root == null) return 0;
        int leftdepth = depth(root.left);
        int rightdepth = depth(root.right);
        
        return leftdepth > rightdepth ? leftdepth + 1 : rightdepth + 1;
    }
}

1.2 从底到顶

思路：

从底到顶，O(N)，进阶
（1）建立辅助方法height求此时平衡树的高度
（2）若root为空，认为平衡，返回true
（3）height方法，返回值为当前root平衡树的高度，若root为空返回0
（4）递归获取左右子树的高度，若任一子树（左右）高度为-1或左右子树差的绝对值>1，则返回-1代表不平衡
否则返回最高平衡子树+1
（5）返回 若root平衡树高度>=0为true

代码：

class Solution {
    public boolean isBalanced(TreeNode root) {
        if(root == null) return true;
        return height(root) >= 0;
    }
    public int height(TreeNode root){
        if(root == null) return 0;

        int lefth, righth;
        if((lefth = height(root.left)) == -1 || (righth = height(root.right)) == -1 || Math.abs(lefth -  righth) > 1) return -1;
        return Math.max(lefth, righth) + 1;
    }
}