【打卡】牛客网：BM36 判断是不是平衡二叉树

最新推荐文章于 2024-09-03 20:17:53 发布

初霁i

最新推荐文章于 2024-09-03 20:17:53 发布

阅读量85

点赞数 1

文章标签： java 算法开发语言

本文链接：https://blog.csdn.net/weixin_47173826/article/details/134282635

版权

自己写的：

每个结点都递归地去判断深度，时间复杂度应该大于o(n)。

比较好的一点：主函数递归。

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param pRoot TreeNode类 
     * @return bool布尔型
     */
    int Height(TreeNode* pRoot){
        if(pRoot == NULL)
            return 0;
        return max(Height(pRoot->left),Height(pRoot->right))+1;
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        // write code here
        if(pRoot == NULL)
            return true;
        if(abs(Height(pRoot->left)-Height(pRoot->right))<=1 && IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right)) 
            return true;
        else
            return false;
    }
};

在递归计算树的深度的时候，顺便判断一下是否是平衡树。保证树的每个结点遍历一次，满足o(n)。

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param pRoot TreeNode类 
     * @return bool布尔型
     */
    int flag = 0;
    int Judge(TreeNode* pRoot){
        if(pRoot == NULL)
            return 0;
        
        int hLeft = Judge(pRoot->left);
        int hRight = Judge(pRoot->right);
        
        if(abs(hLeft-hRight) > 1)
            flag = 1;

        return max(hLeft, hRight)+1;
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        // write code here
        Judge(pRoot);
        if(flag == 0)
            return true;
        else
            return false;
    }
};