判断是否是二叉平衡树

Posted 2020-08-05

这是剑指offer的第39题扩展，是王道程序员求职宝典的P226一道题，其最后一种解法存在错误

下面给出修正后的答案

class Solution {
public:
    int maxDepth(TreeNode* root){
        if(root==NULL)
            return 0;
 
        return 1+max(maxDepth(root->left),maxDepth(root->right));
    }
    bool danbian(TreeNode *root)
        {
        if(root==NULL)
            return 1;
        if(root->left!=NULL && root->right!=NULL)
            return 0;
        if(root->left==NULL)
            ;//danbian(root->right);
        if(root->right==NULL)
            ;//danbian(root->left);
        return 1;
    }
    int minDepth(TreeNode* root){
        
        if(root==NULL)
            return 0;

//        0
//      /  //     1     3
//    /       \ 
//   2          6
         //左子树虽然为空，如上图3，但未走到左右都是叶子节点的节点，不能结算深度，沿另一个分支继续走
        //同理，右子树为空
        
         if(root->left == NULL) return minDepth(root->right) + 1;
         if(root->right == NULL) return minDepth(root->left) + 1;

        int leftDepth = minDepth(root->left);
        int rightDepth = minDepth(root->right);
        return leftDepth < rightDepth ? (leftDepth + 1) : (rightDepth + 1);
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        int max=maxDepth(pRoot);
        int min=minDepth(pRoot);
        //平衡二叉树可能是单边的，而上面的情况假设是左右子树都存在
        if(danbian(pRoot))
            min=1;
        if(max-min>1)
            return 0;
        return 1;
        
        
    }
};