算法笔记-平衡二叉树

判断平衡二叉树

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

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

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

思路：
  平衡二叉树满足当且仅当一棵树的所有子树都是二叉平衡树，所以可以递归判断。

  方法一：自顶向下的递归（前序遍历）：

  对于当前遍历的节点，计算其左右子树的高度，如果高度差不超过 1，再分别递归地遍历左右子节点（注：即使左右子树高度差不超过1，子树仍可能不是平衡二叉树），并判断左子树和右子树是否平衡。这是一个自顶向下的递归的过程。
代码实现：

class Solution {
    public boolean isBalanced(TreeNode root) {
        if (root == null) return true;
        //前序地判断是否平衡
        //如果两棵子树高差不大于1, 再递归的判断子树的子树
        return Math.abs(height(root.left) - height(root.right)) <= 1 && isBalanced(root.left) && isBalanced(root.right);
    }
    //自顶向下计算树的高度
    public int height(TreeNode root) {
        if (root == null) return 0;
        //后序地获取结点高度
        return Math.max(height(root.left), height(root.right)) + 1;
    }
}

时间复杂度：O（n²）

​  方法二：自底向上的递归（后序遍历）

​  在方法一中，对于每个结点，函数height会被调用n次，如果采用自底向上的方法，每个结点只会调用一次height函数：
  对于当前遍历的节点，先递归地判断其左右子树是否平衡，再判断以当前节点为根的子树是否平衡。如果一棵子树是平衡的，则返回其高度（高度一定是非负整数），否则返回 −1。如果存在一棵子树不平衡，则整个二叉树一定不平衡，一路返回-1。
代码实现：

class Solution {
    public boolean isBalanced(TreeNode root) {
        //返回值不是-1就是平衡的
        return height(root) >= 0;
    }
    //自底向上计算树的高度
    public int height(TreeNode root) {
        if (root == null) return 0;
        int leftHeight = height(root.left);
        int rightHeight = height(root.right);
        //某一子树不平衡了就返回-1，一直返回到顶
        if (leftHeight == -1 || rightHeight == -1 || Math.abs(leftHeight - rightHeight) > 1) 
        	return -1;
        //返回树的高度
        return Math.max(leftHeight, rightHeight) + 1;
    }
}

​时间复杂度：O（n）