牛客网 平衡二叉树

题目描述

输入一棵二叉树，判断该二叉树是否是平衡二叉树。

 

话说这题有两种做法。首先平衡二叉树指的是：一棵空树或它的左右两个子树的高度差的绝对值不超过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;
         
    }
     
};