lc124. 二叉树中的最大路径和 python题解

367 阅读1分钟

题目表述:

路径 被定义为一条从树中任意节点出发,沿父节点-子节点连接,达到任意节点的序列。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点,且不一定经过根节点。

路径和 是路径中各节点值的总和。

给你一个二叉树的根节点 root ,返回其 最大路径和 。

image.png

image.png

题目思路:

以root节点为根节点: 二叉树中的最大路径和即为= 左孩子的最大和 + 右孩子的最大和 + 根节点的val

引入辅助函数helper(root),返回单分支的最大和 return max(L,R) + root.val

L = max(self.helper(root.left),0) 和0取最大值是为了避免碰到负数情况

R = max(self.helper(root.right),0)

代码参考:

# 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 maxPathSum(self, root: TreeNode) -> int:
        self.ans = float("-inf")
        self.helper(root)
        return self.ans
    def helper(self,root):
        if root is None:
            return 0
        L = max(self.helper(root.left),0)
        R = max(self.helper(root.right),0)
        self.ans = max(self.ans,L + R + root.val)
        return max(L,R) + root.val