题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
话说这题有两种做法。首先平衡二叉树指的是:一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。
所以直接来想的话,只要保证根节点的左右子节点都是平衡二叉树,并且两个子树的高度相差都是小于等于1就可以了。代码如下所示:
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
if(pRoot==NULL) return true;
return (abs(findheight(pRoot->left) - findheight(pRoot->right))<=1)
&& IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
}
int findheight(TreeNode* pRoot){
if(pRoot==NULL) return 0;
return 1+ max(findheight(pRoot->left), findheight(pRoot->right));
}
};
那么这个方法有一个问题,效率太低。这是因为对于每一个节点都要向下递归寻找它的两个子节点的高度,以及判断是不是平衡二叉树。有没有一次遍历就能成功的做法呢?实际上是有的,也就是后序遍历。
我们先遍历一个根节点的左右两个节点,分别判断这两个子节点的高度差以及判断两点是不是平衡树。这样后续遍历,从下岛上遍历,只需要遍历一次就可以判断出来最终这棵树是不是二叉平衡树了。
代码如下:
class Solution {
public:
bool IsBalanced_Solution(TreeNode* pRoot) {
if(pRoot==NULL) return true;
int init = 0;
return isbalanced(pRoot, &init);
}
bool isbalanced(TreeNode* pRoot, int *p){
if(pRoot == NULL){
*p = 0;
return true;
}
int left, right;
if(isbalanced(pRoot->left, &left) && isbalanced(pRoot->right, &right)){
if(abs(left - right) <=1){
*p = 1 + max(left, right);
return true;
}
else return false;
}
return false;
}
};