剑指offer 判断二叉树是否是平衡二叉树 C++

题目描述

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

在这里，我们只需要考虑其平衡性，不需要考虑其是不是排序二叉树

判断二叉树平衡的条件是

左右子树的深度不能大于1

思路

左右子树都先求深度，再比较深度差是否大于1
因为在遍历每个结点时都要求其左右子树的深度，因此复杂度是O(n^2)
Tips 求二叉树深度 | 如何创建节点 | 完整代码

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(!pRoot) return true;
        if(abs(TreeDeep(pRoot->left) - TreeDeep(pRoot->right) )> 1) return false;
        return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
    }
    // 求二叉树深度
    int TreeDeep(TreeNode* pRoot) {
        if(!pRoot) return 0;
        return 1 + max(TreeDeep(pRoot->left),TreeDeep((pRoot->right)));
    }
};