给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
思路:判断高度是否平衡,后序遍历,左右中;用返回-1的方式表达此时该节点已经不为平衡二叉树,注意这里获取高度时,如果为平衡二叉树,返回值为该节点目前的高度。
class Solution {
public:
//后序遍历 左右中
int getHeight(TreeNode* node){
if(node==NULL)
return 0;
int leftHeight=getHeight(node->left);
if(leftHeight==-1)
return -1;
int rightHeight=getHeight(node->right);
if(rightHeight==-1)
return -1;
if(abs(leftHeight-rightHeight)>1)
return -1;
return 1+max(leftHeight,rightHeight);
}
bool isBalanced(TreeNode* root) {
if(root==NULL)
return true;
int result=getHeight(root);
return result>0?1:0;
}
};