vlambda博客
学习文章列表

2022.4.16

剑指 Offer 55 - II. 平衡二叉树

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

示例 1:

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

    3
   / \
  9  20
    /  \
   15   7

返回 true 。

题解：

平衡二叉树的判定，同样是结合递归思想，将问题转化为一般性的子问题，一个平衡二叉树，那么它的左右子树也是平衡二叉树，且左右子树深度相差不超过1

class Solution {
    public boolean isBalanced(TreeNode root) {
        if (root == null) {
            return true;
        }
        else {
            // 一个平衡二叉树，那么它的左右子树也是平衡二叉树，且左右子树深度相差不超过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;
        } else {
            return Math.max(height(root.left), height(root.right)) + 1;
        }
    }
}