题目描述:

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

38、平衡二叉树_平衡二叉树

解题思路:

在遍历树的每个结点的时候,调用函数TreeDepth得到它的左右子树的深度。如果每个结点的左右子树的深度相差都不超过1,按照定义它就是一棵平衡的二叉树。一边遍历一边判断是否是平衡二叉树

38、平衡二叉树_子树_02

Demo:

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        return TreeDepth(pRoot) != -1;
    }
    int TreeDepth(TreeNode* pRoot){
        if (pRoot == nullptr) return 0;
        int nleft = TreeDepth(pRoot->left);
        // 如果当前结点左子树的高度为-1,则说明当前结点不满足要求
        if (nleft == -1) return -1; 
        int nright = TreeDepth(pRoot->right);
        // 如果当前结点右子树的高度为-1,则说明当前结点不满足要求
        if (nright == -1) return -1;
        // 先判断当前结点是否满足平衡二叉树的要求,不满足直接返回高度为-1
        return (abs(nleft - nright) > 1) ? -1 : max(nleft, nright) + 1;
    }
};