1,题目要求
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
2,题目思路
对于这道题,要求判断一棵树是否是平衡二叉树。
所谓平衡二叉树,即所有的左右子树的高度差小于等于1的树。
简单的办法,是递归地对整颗树的每一个节点进行遍历,判断左右字数的高度之差即可。
这种做法有很明显的问题,在判断上层结点的时候,会多次重复遍历下层结点,增加了不必要的开销。
如果改为从下往上遍历——即在递归的过程中进行判断,如果子树是平衡二叉树,则返回子树的高度;如果发现子树不是平衡二叉树,则直接停止遍历,这样至多只对每个结点访问一次。
在这道题中,我们定义-1为非平衡二叉树的标识。
3,代码实现
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
return balanceDepth(pRoot)!=-1;
}
private:
int balanceDepth(TreeNode* root){
if(!root)
return 0;
int left = balanceDepth(root->left);
if(left == -1)
return -1;
int right = balanceDepth(root->right);
if(right == -1)
return -1;
return abs(left-right)>1?-1:1+max(left,right);
}
};