Leetcode 543. 二叉树的直径

176 阅读1分钟

“Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情。”

今天写了二叉树的中序遍历,所以打算把热题100中关于二叉树的题,都解一遍,方便大家专项学习

一、题目描述

  • 给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

二、思路分析

  • 首先很容易想到,应该是从一个叶子节点到另一个叶子节点

  • 直径长度是节点数减一

  • 计算高度,找到最底下的叶子节点

  • 可以得出直径就是左子树的高度加上右子树的高度加上根节点减一

三、AC 代码:

class Solution {
    //结果变量
    private int res = 0;
    public int diameterOfBinaryTree(TreeNode root) {
        //边界条件
        if (root == null || (root.left == null && root.right == null)) {
            return 0;
        }
        //求出最大直径
        maxDepth(root);

        return res;
    }

    private int maxDepth(TreeNode root) {
        //边界条件
        if (root == null) {
            return 0;
        }
        //左子树的高度
        int leftDepth = maxDepth(root.left);
        //右子树的高度
        int rightDepth = maxDepth(root.right);
        //更新最大值
        res = Math.max(res, leftDepth + rightDepth);
        //树的高度等于子树加上根节点
        return Math.max(leftDepth, rightDepth) + 1;
    }
}

四、总结:

  • 好了,代码撸完,我们总结一下
    • 获取树的高度

    • 直径等于左子树的高度加上右子树的高度

    • 别忘记加一

最后这个算法是Leetcode的第543题,是热题100里的题,去年我刷了热题100用时一个月左右,接下来我会继续更新,小伙伴可以点赞关注,如果你也在刷热题100的话,希望可以对你有一些启发!