剑指Offer(牛客版)--面试题55(2):二叉树平衡树

题目描述：

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

 

分析：

 

完整代码：

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        //声明一个变量，表示高度
        int pDepth = 0;
        //调用函数判断是否为平衡子树
        return IsBalanced(pRoot, &pDepth);

    }
private:
    bool IsBalanced(TreeNode* pRoot, int* pDepth)
    {
        //如果当前节点为空
        if(pRoot == nullptr)
        {
            *pDepth = 0;
            return true;
        }
        //声明两个变量 left ，right；表示左右子树的高度
        int left, right;
        //递归调用函数判断左右子树是否为平衡树，并计算左右子树的高度
        if(IsBalanced(pRoot->left,&left) && IsBalanced(pRoot->right,&right))
        {
            //计算左右子树的高度差
            int value = left - right;
            //如果高度差在[-1,1]
            if(value <= 1&& value >= -1)
            {
                *pDepth = 1 + ((left > right)? left : right);
                return true;
            }
        }
        //如果左右都不为平衡树
        return false;
    }
};