LeetCode-热题100：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 * 10⁴]
• -1000 <= Node.val <= 1000

代码及注释

func maxPathSum(root *TreeNode) int {
    // 初始化一个非常小的值，用于保存最大路径和
    res := math.MinInt32

    // 定义一个递归函数，计算当前节点为根的子树的最大路径和
    var dfs func(*TreeNode) int
    dfs = func(root *TreeNode) int {
        // 如果当前节点为空，返回0
        if root == nil {
            return 0
        }

        // 计算左子树的最大路径和，如果小于0，则取0
        left := max(0, dfs(root.Left))
        
        // 计算右子树的最大路径和，如果小于0，则取0
        right := max(0, dfs(root.Right))

        // 更新全局最大路径和
        res = max(res, root.Val + left + right)

        // 返回当前节点为根的子树的最大路径和
        return root.Val + max(left, right)
    }

    // 调用递归函数，开始计算最大路径和
    dfs(root)

    // 返回最大路径和
    return res
}

代码解释

• 对于当前节点，有四种情况：

  1. 只取左子树路径。
  2. 只取右子树路径。
  3. 取左子树路径 + 当前节点。
  4. 取右子树路径 + 当前节点。

• 对于每个节点，我们都尝试计算这四种情况，并更新全局的最大路径和res。

• 最后，res会存储全局最大路径和。