题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
此题有解法一:依次从根节点开始,判断所有节点的左右子树是否平衡,但这样一来,排在底层的节点可能要被多次重复遍历,所以此方法复杂度高,不推荐。
解法二:后序遍历,左–>右–>根,这样一来,可以“一边遍历,一边判断”,且如果前期就判断为false,那么后面就不用判断了。
后序遍历的精髓:在递归中添加临时变量,深刻体会。
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
int depth;
return IsBalanced(pRoot,&depth);
}
bool IsBalanced(TreeNode* pRoot,int*pDepth){
if(pRoot==NULL){
*pDepth=0;
return true;
}
int leftdepth,rightdepth; //在递归中声明临时变量,用于求父节点传来的指针对应的值。
if(IsBalanced(pRoot->left,&leftdepth)&&IsBalanced(pRoot->right,&rightdepth)){
if(abs(leftdepth-rightdepth)<=1){
*pDepth=leftdepth>rightdepth?leftdepth+1:rightdepth+1;
return true;
}
}
return false;
}
};