平衡二叉树判断

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

解法一：递归实现(需要重复遍历节点多次)

import java.util.*;

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}*/
public class CheckBalance {
    public boolean check(TreeNode root) {
        // write code here
        if(root==null){
            return true;
        }
        int lh=getDepth(root.left);
        int rh=getDepth(root.right);
        int diff=lh-rh;
        if(diff>1||diff<-1){
            return false;
        }
        return check(root.left)&&check(root.right);
    }
    //返回一个节点的深度，即该节点到叶子节点的路径最大长度
    private int getDepth(TreeNode root){
        if(root==null){
            return 0;
        }
        int lh=getDepth(root.left);
        int rh=getDepth(root.right);

        return lh>rh?(lh+1):rh+1;
    }
}

解法二：

import java.util.*;
 
/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}*/
public class CheckBalance {
    public boolean check(TreeNode root) {
        //定义一个引用类型的数据作为平衡标记,通过传引用的方式在递归左右子树时修改平衡标记
        boolean[] res=new boolean[1];
        //从根节点开始遍历树，遍历过程中修改平衡标记
        res[0]=true;
        postCheck(root,1,res);
         
        return res[0];
    }
    public int postCheck(TreeNode root,int depth,boolean[] res){
        if(root==null){
            return depth;
        }
        //遍历一次左子树，获取深度(深度已经在参数改变了，目的是为了检查左子树是否平衡)
        //若遍历左子树过程中修改了平衡标记为false，则子树非平衡，所以当前结点为根的子树非平衡，不再递归，直接返回

        int left_depth=postCheck(root.left,depth+1,res);
        if(res[0]==false){
            return depth;
        }
        //若左子树是平衡的，则遍历右子树并获取深度
        //若遍历右子树过程中修改了平衡标记为false，则子树非平衡，所以当前结点为根的子树非平衡，不再递归，直接返回
        int right_depth=postCheck(root.right,depth+1,res);
        if(res[0]==false){
            return depth;
        }
         
        //若左右子树都是平衡的，则对左右子树深度进行比较，判断当前结点为根的子树是否平衡
        if(Math.abs(left_depth-right_depth)>1){//高度差大于1，当前子树不平衡,修改平衡标记
            res[0]=false;
        }
        //用左右子树深度最大者作为自己的高度
        return Math.max(left_depth,right_depth);
    }
}