题解 | #判断是不是平衡二叉树#

判断是不是平衡二叉树

https://www.nowcoder.com/practice/8b3b95850edb4115918ecebdf1b4d222

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
#include <cstddef>
#include <ios>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param pRoot TreeNode类 
     * @return bool布尔型
     *
     *
     **/
    

    int height(TreeNode * root ){
        if(root == NULL){
            return 0;
        }
        return max(height(root->left),height(root->right)) + 1;
    }

    bool IsBalanced_Solution(TreeNode* pRoot) {
        // write code here
        if(!pRoot){
            return true;
        }
        int left_height = height(pRoot->left);
        int right_height = height(pRoot->right);

        if(left_height - right_height < -1 || left_height - right_height > 1){
            return false;
        }
        return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
    }
};

首先通过递归计算出根节点左子树的高度和根节点右子树的高度,判断根节点的左子树和右子树高度差是否小于1或者大于-1.

然后判断根节点的左子树和根节点的右子树是否是平衡二叉树即可。

全部评论

相关推荐

10-10 16:30
济宁学院 Java
一表renzha:面试官:蓝桥杯三等奖?你多去两次厕所都能拿二等吧
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务