Leetcode110. 平衡二叉树

Leetcode110. 平衡二叉树

题目：
给定一个二叉树，判断它是否是高度平衡的二叉树。
本题中，一棵高度平衡二叉树定义为：
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。
示例 1:

给定二叉树 [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7
返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]

       1
      / \
     2   2
    / \
   3   3
  / \
 4   4
 返回 false 。

题解：

scala代码如下：

/**
    * 迭代
    *
    * @param root
    * @return
    */
  def isBalanced(root: TreeNode): Boolean = {
    depth(root) != -1
  }
  def depth(root: TreeNode): Int = {
    if (root == null) {
      0
    } else if (depth(root.left) == -1) {
      -1
    } else if (depth(root.right) == -1) {
      -1
    } else {
      if (Math.abs(depth(root.left) - depth(root.right)) < 2) {
        Math.max(depth(root.left), depth(root.right)) + 1
      } else {
        -1
      }
    }
  }

java代码：

 /**
     * 迭代
     *
     * @param root
     * @return
     */
    public static boolean isBalanced(TreeNode root) {
        if (root == null) return true;
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()) {

            TreeNode node = stack.pop();
            if (node != null) {
                TreeNode left = node.left;
                TreeNode right = node.right;
                if (Math.abs(height(left) - height(right)) > 1) {
                    return false;
                }
                stack.push(left);
                stack.push(right);
            }
        }
        return true;
    }

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