【剑指Offer】No.27 平衡二叉树

题目描述

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

平衡二叉树：平衡二叉树的左右子树也是平衡二叉树，那么所谓平衡就是左右子树的高度差不超过1.

解题思路

这道题从定义出发，检查每一个左子树与右子树的高度差。所以基于求二叉树的深度，想到每一次的左右子树高度进行检查，并且返回左右子树较大高度作为其根节点所在子树的高度。

public class Solution {
    boolean flag = true;
    public boolean IsBalanced_Solution(TreeNode root) {
        if (root == null) {
            return flag;
        }
        func(root);
        return flag;
    }
    private int func(TreeNode root) {
        if (root == null) {
            return 0;
        }
        
        int left = func(root.left) + 1;
        int right = func(root.right) + 1;
        if (left - right > 1 || right - left > 1) {
            flag = false;
        }
        
        return (left > right) ? left : right;
    }
}