剑指offer 39. 是否为平衡二叉树

39. 是否为平衡二叉树

题目描述

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

任意结点的左右子树高度差不大于1就是平衡二叉树。

C++解法

 class Solution {
 public:
     bool flag = true;    // 记录是否为平衡二叉树，不是则为false
     bool IsBalanced_Solution(TreeNode* pRoot) {
         preOrder(pRoot);
         return flag;
     }
     // 前序递归遍历
     int preOrder(TreeNode* pRoot){
         if(pRoot && flag == true){
             int leftH = preOrder(pRoot->left);
             int rightH = preOrder(pRoot->right);
             if(abs(leftH - rightH) > )
                 flag = false;
             return ((leftH >= rightH) ? (leftH + ) : (rightH + ));
         }
         return ;
     }

 };

Java解法

 public class Solution {
     public boolean IsBalanced_Solution(TreeNode root) {
         if(preOrder(root) == -)
             return false;
         return true;
     }

     // 如果是返回-1， 则表示不是平衡二叉树
     public int preOrder(TreeNode root){
         if(root == null)
             return ;
         int leftH = preOrder(root.left);
         if(leftH == -)
             return -;        //如果发现子树不平衡之后就没有必要进行下面的高度的求解了
         int rightH = preOrder(root.right);
         if(- == rightH)       // 如果发现子树不平衡之后就没有必要进行下面的高度的求解了
             return -;
        if((leftH - rightH) >  || (leftH - rightH < -))
            return -;
         return ((leftH >= rightH) ? leftH : rightH) + ;
     }
 }