关闭

【刷题之路】判断平衡二叉树

标签: 二叉树遍历递归
154人阅读 评论(0) 收藏 举报

采用递归的方式,不断返回左右子树中较大的值作为当前节点的depth,当前节点又与其他一个节点作为上一节点的左右子树,一旦某一左右子树递归叠加深度差大于一,以后全部返回false

class CheckBalance {

public:
    bool check(TreeNode* root) {
        // write code here
        int depth=0;
        return helper(root,depth);
    }
     bool helper(TreeNode* pRoot, int &Depth){
        if (pRoot == NULL){
            Depth = 0;  //递归结束条件,开始返回值
            return true;
        }
 
        int left, right;
        if (helper(pRoot->left, left) && helper(pRoot->right, right)){  //只有当之前返回值全为ture时,叠加才在继续
            Depth = max(left, right) + 1;  //当前节点之下的左右子树的较大深度为当前节点深度
            int dif = left - right;  //比较当前递归节点的左右子树差
            if (dif <= 1 && dif >= -1){
                return true;  //如果条件不成立,则返回之后的false,并且之后的if条件均不执行,一直返回false到结束
            }
        }
 
        return false;
    }
};
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:10410次
    • 积分:799
    • 等级:
    • 排名:千里之外
    • 原创:72篇
    • 转载:2篇
    • 译文:0篇
    • 评论:1条