python leetcode 124. Binary Tree Maximum Path Sum

最新推荐文章于 2024-08-14 09:54:52 发布

Neekity

最新推荐文章于 2024-08-14 09:54:52 发布

阅读量582

点赞数

分类专栏： leetcode python 文章标签： python leetcode

本文链接：https://blog.csdn.net/Neekity/article/details/84978154

版权

python 同时被 2 个专栏收录

263 篇文章 0 订阅

订阅专栏

leetcode

221 篇文章 2 订阅

订阅专栏

又是一道好题啊。难点在于如何根据题意设计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]