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

摸鱼的快乐你不懂

于 2024-07-25 22:36:47 发布

阅读量418

点赞数 3

文章标签： leetcode java

本文链接：https://blog.csdn.net/wudi6688/article/details/140701588

版权

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) {
        // 基本情况：节点为空时贡献值为 0
        if (node == null) {
            return 0;
        }

        // 递归计算左子节点的最大贡献值
        // 只有当左子节点的最大贡献值大于 0 时，才会被选取
        int leftGain = Math.max(maxGain(node.left), 0);
        // 递归计算右子节点的最大贡献值
        // 只有当右子节点的最大贡献值大于 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);
    }
}

代码解释

maxPathSum 方法：
- 入口方法，调用 maxGain 方法计算最大路径和，返回全局变量 maxSum。
maxGain 方法：
- 递归方法，计算当前节点的最大路径贡献值。
- 基本情况：如果当前节点为空，返回贡献值 0。
- 计算左子树和右子树的最大贡献值（只有当贡献值大于 0 时才会选取）。
- 计算包含当前节点的路径和，并更新 maxSum。
- 返回当前节点的最大贡献值（包括该节点值及其左右子树的最大贡献值）。