LeetCode 110. 平衡二叉树

最新推荐文章于 2024-06-09 21:18:28 发布

原创最新推荐文章于 2024-06-09 21:18:28 发布 · 185 阅读

0 ·

CC 4.0 BY-SA版权

算法题_树专栏收录该内容

10 篇文章

订阅专栏

本文探讨了如何判断一个二叉树是否为高度平衡的二叉树，即任意节点的左右子树高度差不超过1。提供了自顶向下的递归算法实现，并分析了其时间复杂度为O(n)，空间复杂度为O(n)。

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点的左右两个子树的高度差的绝对值不超过1。

示例 1:

给定二叉树 [3,9,20,null,null,15,7]
3
/ \
9 20
/ \
15 7
返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]
1
/ \
2 2
/ \
3 3
/ \
4 4
返回 false 。

tips : 自顶向上法, 时间复杂度O(n), 空间复杂度O(n)（递归栈）

class Solution {
public:
    bool isBalancedRecursive(TreeNode* root, int & height) {
        int left_height = 0;
        int right_height = 0;
        if(root == NULL) {
            return true;
        }
        if(isBalancedRecursive(root->left, left_height) && isBalancedRecursive(root->right, right_height) && abs(left_height - right_height) <= 1 ) {
            height = max(left_height, right_height)+1;
            return true;
        } else {
            height = max(left_height, right_height)+1;
            return false;
        }
    }
    bool isBalanced(TreeNode* root) {
        int height = 0;
        return isBalancedRecursive(root, height);
    }
};