124. 二叉树中的最大路径和

题目-困难难度

二叉树中的 路径 被定义为一条节点序列，序列中每对相邻节点之间都存在一条边。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点，且不一定经过根节点。

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

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

示例

示例 1：

输入：root = [1,2,3]
输出：6
解释：最优路径是 2 -> 1 -> 3 ，路径和为 2 + 1 + 3 = 6

示例 2：

输入：root = [-10,9,20,null,null,15,7]
输出：42
解释：最优路径是 15 -> 20 -> 7 ，路径和为 15 + 20 + 7 = 42

提示：

• 树中节点数目范围是 [1, 3 * 104]
• -1000 <= Node.val <= 1000

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/summary-ranges
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

1. dfs

时间
96ms
击败 31.10%使用 Python3 的用户
内存
23.76MB
击败 75.72%使用 Python3 的用户

# 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: Optional[TreeNode]) -> int:
        # 递归获取每一节点的路径和
        def maxGain(node: Optional[TreeNode], maxSum: int) -> int:
            # 如果当前节点为空, 返回0
            if not node:
                return 0
            # 获取左节点路径和
            leftGain = max(maxGain(node.left, maxSum), 0)
            # 获取右节点路径和
            rightGain = max(maxGain(node.right, maxSum), 0)
            # priceNewpath 为当前节点的路径和
            priceNewpath = node.val + leftGain + rightGain
            # 与最大路径和对比,如果大于则替换
            maxSum[0] = max(maxSum[0], priceNewpath)
            # 更新节点值
            return node.val + max(leftGain, rightGain)
        # 创建列表方便更改最大值
        maxSum = [float("-inf")]
        # 从root开始, maxSum作为初始最大路径和
        maxGain(root, maxSum)
        # 返回最大值
        return maxSum[0]