二叉平衡树

题目描述：

输入一棵二叉树，判断该二叉树是否是平衡二叉树。

基本知识：

平衡二叉搜索树（Self-balancing binary search tree）又被称为AVL树。

且具有以下性质：它是一 棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。

class Solution {
    int ldep = 0;
    int rdep = 0;
public:
    int TreeDepth(TreeNode* pRoot){
        if(!pRoot) return 0 ;
            return max(1+TreeDepth(pRoot->left), 1+TreeDepth(pRoot->right));
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(!pRoot)
            return true;
        ldep = TreeDepth(pRoot->left);
        rdep = TreeDepth(pRoot->right);
        if(ldep - rdep > 1 || rdep - ldep > 1)
            return false;
        else
        {
            IsBalanced_Solution(pRoot->left);
            IsBalanced_Solution(pRoot->right);
        }
        return true;
    }
};

小黑的代码：

class Solution {
public:
    int IsBalance(TreeNode* pRoot) {
        if(pRoot == NULL) return 0;
        int ldep = IsBalance(pRoot -> left);
        int rdep = IsBalance(pRoot -> right);
        
        if(abs(ldep - rdep) <= 1 && ldep != -1 && rdep != -1) {
            return max(1 + ldep, 1 + rdep);
        }
        else return -1;
    }
    
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(IsBalance(pRoot) == -1) return false;
        else return true;
    }
};

 小黑的改进：

class Solution {
public:
    int IsBalance(TreeNode* pRoot) {
        if(pRoot == NULL) return 0;
        
        int ldep = IsBalance(pRoot -> left);
        if(ldep == -1) return -1;
        int rdep = IsBalance(pRoot -> right);
        if(rdep == -1) return -1;
        
        if(abs(ldep - rdep) <= 1 ) {
            return max(1 + ldep, 1 + rdep);
        }
        else return -1;
    }
    
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(IsBalance(pRoot) == -1) return false;
        else return true;
    }
};