leetcode 543. 二叉树的直径

最新推荐文章于 2024-09-11 17:20:47 发布

喝牛奶的草履虫

最新推荐文章于 2024-09-11 17:20:47 发布

阅读量222

点赞数

分类专栏： leetcode 文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/milk_paramecium/article/details/122181170

版权

leetcode 专栏收录该内容

77 篇文章 1 订阅

订阅专栏

测量每个节点左子树和右子树的最长长度，并记录最长的左子树+右子树长度（直径），因为直径可能出现在任意一个节点上，所以要测量所有节点。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def diameterOfBinaryTree(self, root: TreeNode) -> int:
        self.max = 0
    
        self.depth(root)
        
        return self.max
        
    def depth(self, root):
        if not root:
            return 0
        l = self.depth(root.left)
        r = self.depth(root.right)
        
        self.max = max(self.max, l+r)
        
        return max(l, r) + 1