《剑指offer》系列 平衡二叉树（Java）

链接

牛客：平衡二叉树

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

题目描述

输入一棵二叉树，判断该二叉树是否是平衡二叉树。

思路

平衡二叉树就是二叉树任意结点左子树和右子树的高度差不超过1，在上一题求二叉树高度的基础上再简单修改下就好了。

代码

牛客：

public class Solution {
    public int getDepth(TreeNode root){
        if(root==null)
            return 0;
        return Math.max(1 + getDepth(root.left),1 + getDepth(root.right));
    }
    
    public boolean IsBalanced_Solution(TreeNode root) {
        if(root==null)
            return true;
        if(Math.abs(getDepth(root.left) - getDepth(root.right)) > 1)
			return false;
        return IsBalanced_Solution(root.left)&&IsBalanced_Solution(root.right);
    }
}

LeetCode：

class Solution {
    public boolean isBalanced(TreeNode root) {
        if(root==null){
            return true;
        }
        if(Math.abs(maxDepth(root.left)-maxDepth(root.right))>1){
            return false;
        }
        return isBalanced(root.left)&&isBalanced(root.right);
    }

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