剑指Offer学习-面试题55:二叉树的深度

最新推荐文章于 2022-11-11 23:55:55 发布

原创最新推荐文章于 2022-11-11 23:55:55 发布 · 298 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#面试 #java #剑指Offer

剑指Offer学习专栏收录该内容

55 篇文章

订阅专栏

本文介绍了一种高效算法，用于计算二叉树的深度并判断其是否为平衡二叉树。通过递归方式，算法首先计算树的深度，然后检查左右子树深度差是否超过1来判断平衡性。提供了两种方法：一种是直接递归计算深度并判断，另一种是在遍历过程中同时计算深度和判断平衡性，提高了效率。

	/**
     * 二叉树的深度
     *
     * @param root
     * @return
     */
    public int treeDepth(TreeNode root) {
        if (null == root) return 0;
        int left = treeDepth(root.left);
        int right = treeDepth(root.right);
        return Math.max(left, right) + 1;
    }


    /**
     * 判断一颗树是不是平衡二叉树(暴力)
     *
     * @param root
     * @return
     */
    public boolean isBalanced(TreeNode root) {
        if (null == root) return true;
        if (Math.abs(treeDepth(root.left) - treeDepth(root.right)) >= 2) return false;
        return isBalanced(root.left) && isBalanced(root.right);
    }


    /**
     * 判断一颗树是不是平衡二叉树(遍历一次)
     *
     * @param root
     * @return
     */
    public boolean isBalanced1(TreeNode root) {
        return process(root).b;
    }

    private Result process(TreeNode root) {
        if (null == root) return new Result(0, true);
        Result left = process(root.left);
        Result right = process(root.right);
        if (left.b && right.b) {
            int abs = Math.abs(left.depth - right.depth);
            if (abs >= 2) return new Result(0, false);
            return new Result(Math.max(left.depth, right.depth) + 1, true);
        }
        return new Result(0, false);
    }

    class Result {
        int depth;
        boolean b;

        public Result(int depth, boolean b) {
            this.depth = depth;
            this.b = b;
        }
    }