93 - 平衡二叉树

最新推荐文章于 2021-04-17 22:47:05 发布

原创最新推荐文章于 2021-04-17 22:47:05 发布 · 259 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

lintcode 专栏收录该内容

218 篇文章

订阅专栏

本文介绍了一种用于判断二叉树是否为平衡二叉树的算法实现。该算法不仅检查根节点的左右子树高度差，还递归地检查每个子树的高度差，确保整棵树的平衡状态。

4.25

不能只判断根节点的左右子树的高度差，

每个子树的高度差都应该判断的。

/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */
public class Solution {
    /**
     * @param root: The root of binary tree.
     * @return: True if this Binary tree is Balanced, or false.
     */
    public boolean isBalanced(TreeNode root) {
        // write your code here
        if(root == null){
            return true;
        }
        int a = treeH(root.left);
        int b = treeH(root.right);
        if(Math.abs(a-b)<=1){
            if(isBalanced(root.left)){
                return isBalanced(root.right);
            }
            else{
                return false;
            }
        }
        else{
            return false;
        }
        
    }
    
    public int treeH(TreeNode root){
        if(root == null){
            return 0;
        }
        int a = treeH(root.left);
        int b = treeH(root.right);
        return Math.max(a,b) + 1;
        
    }
}