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

合集 - LeetCode 树(11)

1.687. 最长同值路径2024-09-11

2.124. 二叉树中的最大路径和2024-09-11

3.2385. 感染二叉树需要的总时间2024-09-114.【模板题】- 144. 二叉树的前序遍历2024-09-125.【模板题】- 94. 二叉树的中序遍历2024-09-126.【模板题】- 145. 二叉树的后序遍历2024-09-127.872. 叶子相似的树2024-09-128.LCP 44. 开幕式焰火2024-09-129.404. 左叶子之和2024-09-1210.671. 二叉树中第二小的节点2024-09-1211.662. 二叉树最大宽度2024-09-24

收起

题目链接	124. 二叉树中的最大路径和
思路	“树的直径”&“树形DP”
题解链接	【视频】彻底掌握直径 DP！从二叉树到一般树！（Python/Java/C++/Go）
关键点	经典的树形DP：当前状态为左右儿子与当前节点值之和，返回时注意应当返回 $max(0,v+max(v_l,v_r))$
时间复杂度	$O(n)$
空间复杂度	$O(n)$

代码实现：

class Solution:
    def maxPathSum(self, root: Optional[TreeNode]) -> int:
        answer = -inf
        def dfs(node):
            if node is None:
                return 0
            l_val = dfs(node.left)
            r_val = dfs(node.right)
            nonlocal answer
            answer = max(answer, l_val + r_val + node.val)
            return max(0, node.val + max(l_val, r_val))
        dfs(root)
        return answer