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

题目链接 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
posted @ 2024-09-11 00:37  WrRan  阅读(8)  评论(0编辑  收藏  举报