题目地址(124. 二叉树中的最大路径和)

题目地址(124. 二叉树中的最大路径和)

https://leetcode.cn/problems/binary-tree-maximum-path-sum/

题目描述

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

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

给你一个二叉树的根节点 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

前置知识

公司

• 暂无

思路

关键点

代码

• 语言支持：Python3

Python3 Code:

# 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:
    ans = float('-inf')
    def maxPathSum(self, root: Optional[TreeNode]) -> int:
        def helper(node):
            """
            返回以当前结点为根节点的MaxPath
            1. 根节点必须选择
            2. 左右子树只选择一个（val值较大的那一个）
            当遍历到一个节点时候，我们需要子节点的信息，然后同时结合自身的val来决定要不要选取左右子树
            以及选取的话，要选哪一个，因此整个过程是后序遍历
            """
            if not node: return 0
            l = helper(node.left)
            r = helper(node.right)
            self.ans = max(self.ans,max(l,0)+max(r,0)+node.val)
            return max(l,r,0)+node.val
        helper(root)
        return self.ans

复杂度分析

令 n 为数组长度。

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$