题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
}
};
如果二叉树的每个节点的左子树和右子树的深度不大于1,它就是平衡二叉树。
先写一个求深度的函数,再对每一个节点判断,看该节点的左子树的深度和右子树的深度的差是否大于1.
//后续遍历时,遍历到一个节点,其左右子树已经遍历 依次自底向上判断,每个节点只需要遍历一次
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
getDepth(pRoot);
return isBalanced;
}
int getDepth(TreeNode * pRoot)
{
if(pRoot == NULL)
return 0;
int left = getDepth(pRoot->left);
int right = getDepth(pRoot->right);
if(abs(left-right)>1)
{
isBalanced = false;
}
return left>right? left+1:right+1;
}
private:
bool isBalanced = true;
};