博客
关于我
leetcode【简单】543、二叉树的直径
阅读量:730 次
发布时间:2019-03-21

本文共 1325 字,大约阅读时间需要 4 分钟。

给定一棵二叉树,你需要计算它的直径长度。直径长度定义为任意两个结点之间路径长度的最大值。路径可能穿过根节点,也可能不穿过根节点。

错误的思考方式

初步的错误想法是直接将左右子树的高度相加,以此得到直径长度。然而,这种方法并不总是正确,因为树中可能存在一条路径,这条路径并不经过根节点,而这一条路径的长度可能比左右子树高度之和更长。

正确的思路与递归公式

正确的思路是将每个节点都视为根节点,分别计算这棵以该节点为根的树的直径长度,然后取最大的那个值作为当前树的直径长度。

递归公式为:

  • 直径 = max(左子树的直径, 右子树的直径, 左子树高度 + 右子树高度)

这个递归公式考虑了如果路径穿过根节点以及不穿过根节点的情况。每次递归返回后,更新全局最大直径。因此,正确的直径长度是所有这些情况中的最大值。

实现方法

使用递归遍历整棵树,每次计算当前节点的左右子树的直径长度,同时计算高度,然后比较并维护一个堆的最大值。在递归返回时,更新全局最大直径值。

代码实现

class TreeNode:    def __init__(self, val=0, left=None, right=None):        self.val = val        self.left = left        self.right = rightclass Solution:    def diameterOfBinaryTree(self, root:TreeNode) -> int:        maxd = [0]  # 使用闭包包装,避免变量修改问题        def dfs(node):            if not node:                return 0            left = dfs(node.left)            right = dfs(node.right)            if left + right > maxd[0]:                maxd[0] = left + right            # 返回当前树的高度            return max(left, right) + 1        dfs(root)        return maxd[0]

代码解释

  • TreeNode类:用于定义二叉树的节点,包含值、左子树和右子树属性。
  • Solution类:处理二叉树直径计算的逻辑。
  • diameterOfBinaryTree:函数接收根节点,返回整棵树的直径长度。
  • maxd数组:用于维护全局最大直径。
  • dfs函数:递归计算当前子树的直径和高度。
  • 递归逻辑
    • 基本情况:If node为空,返回0。
    • 递归调用:计算左、右子树的直径和高度。
    • 更新maxd:比较左+右高度和当前maxd,更新最大值。
    • 返回高度:当前子树的高度是左、高右或其一加一,表示树的高度。
  • 主函数:调用dfs函数并返回maxd的值。
  • 总结

    这个递归方法确保每个可能的路径都被考量,无论路径是否经过根节点,最终返回的maxd是正确的直径长度。

    转载地址:http://ptlrz.baihongyu.com/

    你可能感兴趣的文章
    MySQL中的关键字深入比较:UNION vs UNION ALL
    查看>>
    mysql中的四大运算符种类汇总20多项,用了三天三夜来整理的,还不赶快收藏
    查看>>
    mysql中的字段如何选择合适的数据类型呢?
    查看>>
    MySQL中的字符集陷阱:为何避免使用UTF-8
    查看>>
    mysql中的数据导入与导出
    查看>>
    MySQL中的时间函数
    查看>>
    mysql中的约束
    查看>>
    MySQL中的表是什么?
    查看>>
    mysql中穿件函数时候delimiter的用法
    查看>>
    Mysql中索引的分类、增删改查与存储引擎对应关系
    查看>>
    Mysql中索引的最左前缀原则图文剖析(全)
    查看>>
    MySql中给视图添加注释怎么添加_默认不支持_可以这样取巧---MySql工作笔记002
    查看>>
    Mysql中获取所有表名以及表名带时间字符串使用BetweenAnd筛选区间范围
    查看>>
    Mysql中视图的使用以及常见运算符的使用示例和优先级
    查看>>
    Mysql中触发器的使用示例
    查看>>
    Mysql中设置只允许指定ip能连接访问(可视化工具的方式)
    查看>>
    mysql中还有窗口函数?这是什么东西?
    查看>>
    mysql中间件
    查看>>
    MYSQL中频繁的乱码问题终极解决
    查看>>
    MySQL为Null会导致5个问题,个个致命!
    查看>>