leetcode, LC39: 判断二叉树是否为平衡二叉树

最新推荐文章于 2021-02-10 21:22:29 发布

不喜欢敲代码的医生不是一个好Zach

最新推荐文章于 2021-02-10 21:22:29 发布

阅读量52

点赞数

分类专栏： leetcode 文章标签：二叉树算法数据结构 leetcode

本文链接：https://blog.csdn.net/weixin_40883496/article/details/109739389

版权

leetcode 专栏收录该内容

56 篇文章 0 订阅

订阅专栏

1 题目描述

本题要求判断给定的二叉树是否是平衡二叉树
平衡二叉树的性质为: 要么是一棵空树，要么任何一个节点的左右子树高度差的绝对值不超过 1。
一颗树的高度指的是树的根节点到所有节点的距离中的最大值。

3 代码实现

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */

class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @return bool布尔型
     */
    bool isBalanced(TreeNode* root) {
        int depth = 0;
        return isBalancedHelper(root, depth);
    }
    bool isBalancedHelper(TreeNode* root, int &depth){
        if(root == NULL){
            depth = 0;
            return true;
        }
        
        int leftSubtreeDepth, rightSubtreeDepth;
        if(isBalancedHelper(root->left, leftSubtreeDepth) && isBalancedHelper(root->right, rightSubtreeDepth))
            if(abs(leftSubtreeDepth - rightSubtreeDepth) <= 1){
                depth = 1 + max(leftSubtreeDepth, rightSubtreeDepth);
                return true;
            }
        return false;
    }
};