题解 | #判断是不是平衡二叉树#

class Solution {
public:
    // 返回以该节点为根节点的二叉树的高度，如果不是平衡二叉树则返回-1
    int getHeight(TreeNode* node){
        if(!node) return 0; // 如果当前节点为空，返回0
        int leftHeight = getHeight(node->left); // 获得左子树的高度
        if(leftHeight == -1) return -1; // 如果左子树不是平衡二叉树，则整棵树也不是，直接返回-1
        int rightHeight = getHeight(node->right);   // 获得右子树的高度
        if(rightHeight == -1) return -1;    // 同理，如果右子树不是平衡二叉树，则直接返回-1
        // 如果左右子树的高度差大于1，则不是平衡二叉树，返回-1
        // 否则，返回树的高度，包括当前节点，所以要加1
        return abs(leftHeight - rightHeight) > 1 ? -1 : 1 + max(leftHeight, rightHeight);
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        return getHeight(pRoot) == -1 ? false : true;
    }
};

来自《代码随想录》~~~