算法---LeetCode 110. 平衡二叉树(同剑指offer 55-2)

1. 题目

原题链接

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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 。

限制：

0 <= 树的结点个数 <= 10000

注意：本题与主站 110 题相同：https://leetcode-cn.com/problems/balanced-binary-tree/

Related Topics 树 深度优先搜索
👍 153 👎 0

2. 题解

2.1 解法1: 自底向上递归(利用后序遍历)

递归函数作用: 计算左右子树的高度差是否符合平衡条件, 若大于 1 ,说明非平衡,直接返回 -1, 否则返回树高, 即节点 root 的左右子树中最大高度加 1

1. 使用后序遍历, 先计算左子树高度, 再计算右子树高度, 若为-1 直接返回
2. 最后判断当前左右子树高度是否满足要求

    class Solution {
        public boolean isBalanced(TreeNode root) {
            return dfs(root) != -1;
        }

        public int dfs(TreeNode root) {
            if (root == null) return 0;
            int left = dfs(root.left);
            if (left == -1) return -1;
            int right = dfs(root.right);
            if (right == -1) return -1;
            // 此时说明左右子树均平衡, 返回的均是高度
            // 判断是否符合平衡条件, 若符合, 返回高度, 继续向上计算
            return Math.abs(left - right) > 1 ? -1 : Math.max(left, right) + 1;
        }
    }

2.2 解法2: 自顶向下递归(利用先序遍历)

统计每个节点高度, 然后递归判断高度差

    class Solution {
        public boolean isBalanced(TreeNode root) {
            if (root == null) {
                return true;
            }
            int leftH = height(root.left);
            int rightH = height(root.right);
            if (Math.abs(leftH - rightH) > 1) {
                return false;
            } else {
                return 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;
        }
    }

参考:
官方题解