欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

5.判断一棵树是否是平衡二叉树

程序员文章站 2022-05-16 14:56:28
...

平衡二叉树:如果某个二叉树中的任意节点的左右子树的深度相差不超过1。

解题思路:我们可以根据后序遍历的思想来考虑。如果一个节点的左右子树的高度不超过1,并且左右子树的高度值不是-1,我们就把高度值返回上一个节点,否则我们返回值-1。如果最后到根节点的返回值是-1的话我们就判断此二叉树为非平衡二叉树,否则为平衡二叉树。

Java 代码如下:

public class Main{
    public boolean IsBalanced_Solution(TreeNode root) {
        if(root == null)return true;
        if(IsBalanced(root) == -1)return false;
        return true;
    }
    public int IsBalanced(TreeNode root){
        if(root == null) return 0;        
        int left = IsBalanced(root.left);
        int right = IsBalanced(root.right);
        int diff = Math.abs(left - right);
        if(left == -1 || right == -1 || diff > 1)return -1;
        return (left > right)?left+1:right+1;
    }
}