剑指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);//如果平衡,返回树的深度。 } };