Leetcode【二叉树中的最大路径和】

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 * 104]
• -1000 <= Node.val <= 1000

class Solution {
    // 用于存储最终结果的变量，初始化为最小整数值
    int maxSum = Integer.MIN_VALUE;

    // 主方法，计算二叉树中最大路径和
    public int maxPathSum(TreeNode root) {
        maxGain(root); // 递归计算最大路径和
        return maxSum; // 返回最终结果
    }

    // 计算以当前节点为根的子树中的最大路径和
    public int maxGain(TreeNode node) {
        if (node == null) {
            return 0; // 如果节点为空，返回 0
        }
        
        // 递归计算左右子节点的最大贡献值
        // 只有当最大贡献值大于 0 时，才会选取对应子节点
        int leftGain = Math.max(maxGain(node.left), 0);
        int rightGain = Math.max(maxGain(node.right), 0);

        // 当前节点的最大路径和是节点值加上左右子节点的最大贡献值
        int priceNewpath = node.val + leftGain + rightGain;

        // 更新最大路径和的答案
        maxSum = Math.max(maxSum, priceNewpath);

        // 返回当前节点的最大贡献值（用于计算父节点的路径和）
        return node.val + Math.max(leftGain, rightGain);
    }
}

解题思路

1. 定义 maxSum:

   • 这是一个全局变量，用于保存当前遇到的最大路径和。初始化为 Integer.MIN_VALUE 确保可以容纳树中任何路径的和。

2. maxPathSum 方法:

   • 这是主方法，入口点。它调用 maxGain 方法来计算整个树的最大路径和，并返回最终结果。

3. maxGain 方法:

   • 这是一个递归方法，用于计算以当前节点为根的子树中的最大路径和。递归遍历左子树和右子树，并计算每个子树的最大贡献值。

   • 基线条件: 如果节点为空，返回 0（因为空节点不贡献路径和）。

   • 计算左子树和右子树的最大贡献值:

     • 通过递归计算左子树和右子树的最大路径和，并确保贡献值不小于 0。如果贡献值小于 0，则不考虑该子树。

   • 计算当前节点的最大路径和:

     • 当前节点的最大路径和是节点值加上左子树和右子树的最大贡献值。即：priceNewpath = node.val + leftGain + rightGain。

   • 更新全局最大路径和:

     • 比较并更新 maxSum。

   • 返回当前节点的最大贡献值:

     • 当前节点的最大贡献值是节点值加上左子树或右子树的最大贡献值（选取较大的一个）。这用于父节点的路径和计算。