判断一棵树是否是平衡二叉树

题目描述

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

思路

后序遍历二叉树，即先算出左子树的高度，再算出右子树的高度，判断左右子树的高度差，如果小于等于1，则返回根结点的高度（左右子树当中最高的高度加1）；否则此根结点不满足平衡二叉树的条件，终止遍历。

代码（c++）

class Solution {
public:
    int postorder(TreeNode* pRoot){
        if(pRoot==NULL) return 0;
        int left_height=postorder(pRoot->left);
        if(left_height==-1) return -1;
        int right_height=postorder(pRoot->right);
        if(right_height==-1) return -1;
        return abs(left_height-right_height)>1?-1:max(left_height,right_height)+1;
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        return postorder(pRoot)!=-1;
    }
};