【剑指Offer】JZ39平衡二叉树 C++

题目地址：牛客网
题目：

输入一棵二叉树，判断该二叉树是否是平衡二叉树。
在这里，我们只需要考虑其平衡性，不需要考虑其是不是排序二叉树
平衡二叉树（Balanced Binary Tree），具有以下性质：它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。

示例1

输入
{1,2,3,4,5,6,7}
返回值
true

思路：
用之前一题求二叉树深度的代码作为函数，可以求结点的高度。
写了一串判断语句，因为懒得优化了。。
如果二叉树是空的直接返回true。
如果二叉树的根下面只有一个结点，那也是ture，反之如果有一串左子树或者一串右子树就是false。
剩下可以直接看左右子树高度进行判断。

代码：

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