leetcode 110-判断一棵树是否为平衡二叉树 #算法#

2022-06-23 11:24:08 浏览数 (1)

原题如下

Given a binary tree, determine if it is height-balanced. 给出一棵二叉树,判断它是否高度平衡。 For this problem, a height-balanced binary tree is defined as: 高度平衡二叉树定义为:

代码语言:javascript复制
a binary tree in which the depth of the two subtrees of every node never differ by more than 1.
任何节点的两棵子树的深度差不能大于一。

换个说法就是,左右子树的高度(深度)差不能超过一并且左右子树也是平衡二叉树。

思路

首先需要计算一棵树的深度,一棵树的深度为其左右子树的较大深度加上一,左右子树的深度也可以用同样的方法计算出,可以用递归实现,终止条件是根节点为空时深度为0; 有了深度之后,就可以比较某一节点的左右子树的深度差是否小于等于1,并要求左右子树也是平衡树,同样可以用递归实现,终止条件是空树是平衡树。

代码

代码语言:javascript复制
class Solution {
public:
    int depth(TreeNode* root){
        if(root == NULL) return 0;
        int leftDepth = depth(root->left);
        int rightDepth = depth(root->right);
        return (leftDepth > rightDepth ? leftDepth : rightDepth)   1;
    }
    
    bool isBalanced(TreeNode* root) {
        if(root == NULL) return true;
        return abs(depth(root->left) - depth(root->right)) <= 1 
            && isBalanced(root->left) 
            && isBalanced(root->right);
    }
};

0 人点赞