剑指Offer——判断平衡二叉树

最直接做法

最直接的做法就是根据平衡二叉树的定义和特点来判断，1）一棵平衡二叉树的左右子树都是平衡二叉树；2）平衡二叉树的左右子树的高度差不超过1。

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

但是这种做法存在一个问题，就是计算高度时会多次遍历子节点，而且没有剪枝。因此下面有一种剪枝的做法。

经过剪枝后的做法

根据前面提到的特点1可以知道，如果左右子树已经不是平衡二叉树了，则直接层层返回-1，而不用计算剩余的子树了，类似于短路求值的思想。

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        return Depth(pRoot)!=-1;
    }
    
    int Depth(TreeNode* pRoot)
    {
        if(!pRoot)
            return 0;
        int left=Depth(pRoot->left);
        if(left==-1) return -1;
        int right=Depth(pRoot->right);
        if(right==-1) return -1;
        return abs(left-right)>1?-1:max(left, right)+1;
    }
};