leetcode-543 二叉树的直径 Python

最新推荐文章于 2024-03-12 14:57:25 发布

原创最新推荐文章于 2024-03-12 14:57:25 发布 · 297 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode #算法

本文深入探讨了二叉树直径的两种高效算法实现。通过定义深度递归函数，巧妙地在获取节点深度的同时，计算出经过该节点的最长路径长度，从而找到整个二叉树的直径。第一种方法通过在递归过程中更新全局变量来记录最大直径，而第二种方法则是在深度计算中直接返回节点的最长路径长度，并更新最大值。

方法一：

class Solution:
    def diameterOfBinaryTree(self, root: TreeNode) -> int:
        self.ans = 1
        def depth(node):
            if not node: return 0
            L = depth(node.left)
            R = depth(node.right)
            self.ans = max(self.ans, L+R+1)
            return max(L, R)+1
        depth(root)
        return self.ans-1

方法二：

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