python leetcode 124. Binary Tree Maximum Path Sum

最新推荐文章于 2024-03-25 21:39:45 发布

原创最新推荐文章于 2024-03-25 21:39:45 发布 · 621 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode

python 同时被 2 个专栏收录

263 篇文章

订阅专栏

leetcode

221 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索(DFS)算法解决树形结构中寻找最大路径和的问题。通过定义helper函数来计算包含当前节点的最大单边路径和，避免了路径分岔的情况，最终返回整个树的最大路径和。

又是一道好题啊。难点在于如何根据题意设计DFS过程。想一想如何求得由这个节点向下的路径最大值。这个节点的值必然包括，左边的值是max(0,l),右边的是max(0,r)。所以设置helper函数是求包括这个节点在内的最大单边路径和，不能直接设置为最大双边路径和（这样会分岔不符合题意）。

class Solution:
    def maxPathSum(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root: return 0 
        res=[-2**31]
        def helper(root):
            if not root: return 0 
            rv,l,r=root.val,0,0
            if root.left:
                l=helper(root.left)
            if root.right:
                r=helper(root.right)
            res[0]=max(res[0],rv+max(0,l)+max(0,r))
            return rv+max(l,r,0)
        helper(root)
        return res[0]