LeetCode-剑指55-II.平衡二叉树

最新推荐文章于 2024-10-04 20:45:38 发布

犬兄的海角

最新推荐文章于 2024-10-04 20:45:38 发布

阅读量167

点赞数

分类专栏： LeetCode刷题记录文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/weixin_43825194/article/details/127951630

版权

LeetCode刷题记录专栏收录该内容

312 篇文章 1 订阅

订阅专栏

在这里插入图片描述

1、深度遍历

我们可以在上一题LeetCode-剑指55-I.二叉树的深度的基础上继续改进。我们在获得了每一个节点的深度之后，我们比较当前节点的左右子树是否为平衡二叉树以及当前节点的左右子树的深度之差是否小于等于1即可，若是则返回true并继续递归。

class Solution {
public:
    int getDepth(TreeNode *node) {
        if (node == NULL) return 0;
        return 1 + max(getDepth(node->left), getDepth(node->right));
    }

    bool isBalanced(TreeNode *root) {
        if (root == NULL) return true;
        return isBalanced(root->left) && isBalanced(root->right) && abs(getDepth(root->left) - getDepth(root->right)) <= 1;
    }
};