牛客题霸面试真题——二叉树：BM36 判断是不是平衡二叉树

原题链接：https://www.nowcoder.com/practice/8b3b95850edb4115918ecebdf1b4d222

思路

已知叶子结点的高度为1，根节点的高度最大。（下图参考自代码随想录）

要想判断是否时平衡二叉树，就是要左右子树的高度差，那么就要从底向上遍历二叉树，也就是按照先序遍历（中左右）。所以写成递归如下，当节点为空时返回0，只要在遍历的过程中出现了左右子树的高度差>1，就无须再执行函数，直接返回false即可。如果左右子树的高度差<=1，那么返回的是以node为根的树的高度：当前左右子树的最大高度+1

代码

class Solution {
public:
    int getHeight(TreeNode* node){
        if(!node) return 0;
        int leftHeight = getHeight(node->left);
        if(leftHeight==-1) return -1;
        int rightHeight = getHeight(node->right);
        if(rightHeight==-1) return -1;
        
        return abs(leftHeight-rightHeight)>1? -1:1 +max(leftHeight,rightHeight);
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        return getHeight(pRoot)==-1? false:true;
    }
};