题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
-
解题思路
递归判断每个节点的深度是否满足平衡二叉树的要求,即左右子树的高度差不超过1。 -
注意点
递归:从上往下递归会造成重复计算影响效率,可以通过后序遍历从下往上解决。
剪枝:一旦有结点判断出不是平衡二叉树,则不必再判断后续未处理的结点,直接返回结果。 -
解题步骤
递归计算结点的深度,从最左下的结点开始判断,如果不满足平衡二叉树的结点深度要求,直接返回-1表示不符合,否则返回结点的深度。直到递归完毕,返回值都不是-1,则表示是平衡二叉树。
【考点】 树 递归
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
if(getDepth(pRoot) != -1) return true;
return false;
}
int getDepth(TreeNode* pRoot){
if(pRoot == nullptr) return 0;
int left = getDepth(pRoot->left);
if(left == -1) return -1;
int right = getDepth(pRoot->right);
if(right == -1) return -1;
return abs(left - right) > 1 ? -1 : 1+max(left, right);
}
};