LeetCode-热题100：543. 二叉树的直径

最新推荐文章于 2024-08-20 10:34:34 发布

つばめのりゅう℡

最新推荐文章于 2024-08-20 10:34:34 发布

阅读量158

点赞数 1

文章标签： leetcode 算法数据结构 golang 二叉树

本文链接：https://blog.csdn.net/qq_44852180/article/details/137826849

版权

该篇文章介绍了如何使用深度优先搜索（DFS）算法来求解给定二叉树的直径问题，通过递归计算节点的左右子树深度并更新最大直径。示例展示了从节点4到3或5的最长路径长度。

摘要由CSDN通过智能技术生成

题目描述

给你一棵二叉树的根节点，返回该树的直径。

二叉树的直径是指树中任意两个节点之间最长路径的长度。这条路径可能经过也可能不经过根节点 root 。

两节点之间路径的长度由它们之间边数表示。

示例 1：
在这里插入图片描述

输入： root = [1,2,3,4,5]
输出： 3
解释： 3 ，取路径 [4,2,1,3] 或 [5,2,1,3] 的长度。

示例 2：

输入： root = [1,2]
输出： 1

提示：

树中节点数目在范围 [1, 10⁴] 内
-100 <= Node.val <= 100

代码及注释

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */

func diameterOfBinaryTree(root *TreeNode) int {
    // 定义一个内部的 dfs 函数用于深度优先搜索
    var dfs func(*TreeNode) int
    // 初始化最大直径为 0
    res := 0

    dfs = func(node *TreeNode) int {
        // 如果节点为空，返回 0
        if node == nil {
            return 0
        }

        // 递归计算左子树和右子树的深度
        l := dfs(node.Left)
        r := dfs(node.Right)

        // 更新最大直径
        res = max(res, l + r)

        // 返回当前节点的深度
        return max(l, r) + 1
    }
    // 调用 dfs 函数，开始计算
    dfs(root)
    return res
}

// 辅助函数，返回两个整数的最大值
func max(a, b int) int {
    if a > b {
        return a
    }
    return b
}

代码解释

定义 dfs 函数：dfs 函数用于深度优先搜索二叉树，并在递归过程中更新最大直径。
递归计算深度：
- 如果节点为空，则返回 0。
- 递归计算左子树 l 和右子树 r 的深度。
- 更新当前直径的最大值 res 为 l + r。
- 返回当前节点的深度 max(l, r) + 1。
返回结果：最后返回最大直径 res。