题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
首先真的要吐槽下这个剑指offer这题目抄袭leetcode,平衡二叉树,然后实际抄的条件都掉了,人家是“高度平衡的二叉树”!
leetcode对应题目:110. 平衡二叉树 题目解答
总结:
如果是自顶向下的遍历,先判断当前node,在判断左子树,再判断右子树可以不?可以。但是不是好方法,因为你重复计算了很多,相反采取dp的思想,自下向上的解决不就节省了计算的代价。
所以使用后序遍历,先访问左右子树,并且在遍历每个节点的时候记录它的深度,这样在返到根节点访问的时候,我们就不用再重复遍历下面的节点来得到根节点的的深度,直接max(left_depth,right_depth)+1就可以了。
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
if(!pRoot) return true;
int left_dep = getDepth(pRoot->left);
int right_dep = getDepth(pRoot->right);
if(abs(left_dep-right_dep) > 1 ) return false;
return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
}
int getDepth(TreeNode* node){
if(!node) return 0;
return 1+ max(getDepth(node->left),getDepth(node->right));
}
};