leetcode-110 平衡二叉树-平衡二叉树

原创于 2020-11-12 11:13:38 发布 · 197 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #leetcode #算法

leetcode 专栏收录该内容

23 篇文章

订阅专栏

本文介绍了一种判断二叉树是否为平衡二叉树的方法。平衡二叉树定义为任意节点左右子树高度差不超过1的二叉树。通过递归获取每个节点左右子树的深度，并比较深度差来实现判断。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

leetcode-110 平衡二叉树

平衡二叉树要求所有节点的左右子树的高度差小于1，因此，只需在遍历的时候返回其左右子树的深度。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    private boolean flag = true;
    public int checkTree(TreeNode root, int depth){
        if (root == null|| ! flag) return depth;
        int leftDepth = checkTree(root.left, depth + 1);
        int rightDepth = checkTree(root.right, depth + 1);
        // 计算高度差
        if (Math.abs(leftDepth - rightDepth) > 1){
            flag = false;
            return 0;
        }
        // 返回其左右子树最大的深度作为当前节点的深度
        return Math.max(leftDepth, rightDepth);
    }
    public boolean isBalanced(TreeNode root) {
        if (root == null) return true;
        checkTree(root, 0);
        return flag;
    }
}