剑指offer:输入一棵二叉树,判断该二叉树是否是平衡二叉树
思路:核心是用一个函数,既能返回深度,又能返回是否平衡。
IsBalanced_Solution_Core函数既能求树的深度,又可以返回是否是平衡二叉树,如果不平衡,则返回-1
如果节点为空,返回0;递归求左子树的深度和是否平衡,如果左子树不平衡,返回-1;递归求右子树的深度和是否平衡,如果右子树不平衡,返回-1;
如果上面条件都不满足,则求左右子树深度差是否小于等于1,如果不是,则返回-1,否则返回深度。
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
if(IsBalanced_Solution_Core(pRoot) == -1)
return false;
else
return true;
}
int IsBalanced_Solution_Core(TreeNode* pRoot){
if(pRoot == NULL)
return 0;
int nLeft = IsBalanced_Solution_Core(pRoot->left);
if(nLeft == -1)//如果左子树不平衡,返回-1
return -1;
int nRight = IsBalanced_Solution_Core(pRoot->right);
if(nRight == -1)//如果右子树不平衡,返回-1
return -1;
if(nLeft - nRight > 1 || nLeft - nRight < -1 )//如果左右子树深度差大于1,返回-1
return -1;
return 1 + (nLeft > nRight ? nLeft : nRight);//如果平衡,返回树的深度。
}
};