输入一棵二叉树,判断该二叉树是否是平衡二叉树。
平衡二叉树的定义:左右子树节点差不大于1,并且左右子树也都是平衡二叉树
实际上与计算二叉树深度异曲同工,对于每一个节点,需要返回左右子树的最大深度,并且比较差值,并返回左右子树的最大值作为当前节点的深度
代码如下:
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
int dep = 0;
return IsBalanced(pRoot, dep);
}
bool IsBalanced(TreeNode* root ,int &dep){ //需要将每一层的深度不断传递
if(root==NULL) return true;
int left=0,right=0;
if(IsBalanced(root->left,left) && IsBalanced(root->right,right)){ //判断条件
int dif = left – right;
if(dif<-1 || dif >1) //如果差值不满足1之内,则直接返回false
return false;
dep = (left > right ? left : right) + 1; //如果满足,则返回左右子树较大值作为当前节点深度,参与上一节点比较
return true;
}
return false;
}
};