Leetcode-124-二叉树最大路径和-Python

这题表面是"求最大路径和",但难点在于路径不必经过根,而且可以从任意节点走到任意节点。如果只按"根到叶"的思路做,很容易漏掉最优答案。


一、构思:先想"枚举所有路径"可不可行?

直觉上,要找最大路径和,可以枚举所有路径:

  • 二叉树中任意两点之间只有一条简单路径
  • 如果有 n 个节点,所有点对路径数量约 n*(n-1)/2
  • 每条路径再求和 → 总复杂度大概 O(n^2) 甚至更高

结论:理论可行,但太慢(尤其 n=10^4 时直接爆炸)。

所以必须换思路,避免真正枚举所有路径。


二、关键一招:把"路径"拆成两类

这题的关键在于每个节点都可以做"拐点"。

在一个节点上,路径最多只能有三段:

左子树贡献 + 当前节点 + 右子树贡献

但注意:

  • 用于更新答案时,可以"左右都要"(把当前节点当拐点)
  • 返回给父节点时,只能选"左或右的一边",因为路径向上时不能分叉

这就是本题的"关键一招":

> 拐点路径用来更新全局最大值;单边路径用来向上返回贡献。


三、算法设计(一次 DFS 解决)

对每个节点 root 做 DFS,返回"单边最大贡献":

  1. 先拿到左/右子树的最大贡献
  1. 把负贡献当成 0(因为负数只会拉低)
  1. 用 左 + 根 + 右 更新全局最大值
  1. 返回 根 + max(左, 右) 给父节点

四、核心代码

python 复制代码
class Solution:
    def maxPathSum(self, root: Optional[TreeNode]) -> int:
        res = -1e12

        def dfs(root):
            nonlocal res
            if root is None:
                return 0

            leftMax = dfs(root.left)
            rightMax = dfs(root.right)

            # 1) 拐点路径:用于更新全局最大
            curSum = root.val
            if leftMax > 0:
                curSum += leftMax
            if rightMax > 0:
                curSum += rightMax
            res = max(res, curSum)

            # 2) 单边路径:用于向上返回
            singleSum = root.val + max(0, max(leftMax, rightMax))
            return singleSum

        dfs(root)
        return res

五、为什么不能只考虑"拐点路径"?

因为"拐点路径"只能用于更新答案,

但父节点要继续往上拼路径时,必须知道"当前节点能给的单边最大贡献"。

否则无法把路径接上去,结果就不完整。


六、与"枚举所有路径"的关系

你之前问过"枚举所有路径"的思路:

  • 枚举任意两点 → O(n^2)
  • 每条路径计算和 → O(n^3)(如果从头算)
  • 即便用前缀和/LCA优化,也很难降到 O(n)

而这道题用 DFS 的"关键一招",等价于在 O(n) 内隐式枚举了所有路径,但只保留对答案有贡献的部分。


七、复杂度

  • 时间复杂度:O(n)
  • 空间复杂度:O(h)(递归深度)
相关推荐
VALENIAN瓦伦尼安教学设备8 小时前
ASHOOTER激光对中仪如何通过颜色确定调整是否合适
数据库·嵌入式硬件·算法
货拉拉技术8 小时前
重塑 Agent 度量衡:基于 LLM-as-a-Judge 的离线评估体系与实践
算法·设计模式
车压9 小时前
从“为什么”理解注意力机制
算法
wabs6669 小时前
关于哈希表【力扣15.三数之和的思考】
数据结构·算法·leetcode·散列表·哈希表·三数之和
退休倒计时9 小时前
【每日一题】LeetCode 88. 合并两个有序数组 TypeScript
算法·leetcode·职场和发展·typescript
键盘会跳舞9 小时前
C++:函数对象与 std::function 源码级深度拆解——泛型算法的策略内核与可调用对象统一封装
c++·算法·仿函数
手写码匠9 小时前
华为云Flexus+DeepSeek征文|Agent 评测体系实战:用 DeepSeek-R1 当裁判,打造 Dify Agent 的自动化回归测试流水线
人工智能·深度学习·算法·aigc
LONGZETECH9 小时前
无人机实训高成本痛点解法:虚拟仿真实现 70% 耗材损耗下降
大数据·算法·unity·架构·无人机
不会就选b9 小时前
算法日常・每日刷题--<队列,宽搜>4
算法
码字的特恩10 小时前
微软确认暂不为 Windows 11 加入透明效果自定义功能,建议用户使用第三方工具
人工智能·windows·算法·microsoft·计算机·大模型·编程