【LeetCode】二叉树中的最大路径和 [H]（递归）

124. 二叉树中的最大路径和 - 力扣（LeetCode）

一、题目

路径 被定义为一条从树中任意节点出发，沿父节点-子节点连接，达到任意节点的序列。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点，且不一定经过根节点。

路径和 是路径中各节点值的总和。

给你一个二叉树的根节点 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

二、代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public class Info{
        // 从左子树或者右子树延伸出来的连到x节点的最大路径和（注意这个只是从左子树或右子树开始连到x为止的路径，并不是从左子树跨过x再连到右子树的路径）
        public int maxPathSumFromHead;
        // 以当前节点为根节点的整棵树的最大路径和（路径并不一定要过x节点）
        public int maxPathSum;

        public Info(int maxPathSumFromHead, int maxPathSum) {
            this.maxPathSumFromHead = maxPathSumFromHead;
            this.maxPathSum = maxPathSum;
        }
    }

    public int maxPathSum(TreeNode root) {
        // 无效参数
        if (root == null) {
            return 0;
        }

        // 返回root的maxPathSum信息
        return process(root).maxPathSum;
    }

    // 二叉树递归
    public Info process(TreeNode x) {
        // basecase  空的话直接返回null，让上层去做相关的判断
        if (x == null) {
            return null;
        }

        // 通过递归获得左右子树的信息
        Info leftInfo = process(x.left);
        Info rightInfo = process(x.right);

        // 1、先计算maxPathSumFromHead。一共就有三种情况：1)只有x 2）x往左扎 3）x往右扎
        // 先将maxPathSumFromHead初始化为x.val，情况一
        int maxPathSumFromHead = x.val;
        // 每一步操作都要先判断leftInfo和rightInfo是否为空
        // 去比较x往左扎的路径最大值是否是比maxPathSumFromHead大，大则更新maxPathSumFromHead，情况二
        if (leftInfo != null) {
            maxPathSumFromHead = Math.max(maxPathSumFromHead, x.val + leftInfo.maxPathSumFromHead);
        }
        // 去比较x往右扎的路径最大值是否是比maxPathSumFromHead大，大则更新maxPathSumFromHead，情况三
        if (rightInfo != null) {
            maxPathSumFromHead = Math.max(maxPathSumFromHead, x.val + rightInfo.maxPathSumFromHead);
        }

        // 2、再计算以x为根节点的整棵树的最大路径和maxPathSum。
        // 一共就有六种情况：1) 只有x  2)左树整体的最大路径和（不过x节点，也就是左树的maxPathSum） 3) 右树整体的最大路径和（不过x节点，也就是右树的maxPathSum）  4)从左树连接到x节点的最大路径和（只有x到左树的路径）   5)从右树连接到x节点的最大路径和（只有x到右树的路径）  6)从左树连接到x节点，再延伸到右树的最大路径和
        // 先将maxPathSum初始化为x.val，情况一
        int maxPathSum = x.val;
        // 去比较左子树的最大路径和是否是比maxPathSum大，大则更新maxPathSum，情况二
        if (leftInfo != null) {
            maxPathSum = Math.max(maxPathSum, leftInfo.maxPathSum);
        }
        // 去比较右子树的最大路径和是否是比maxPathSum大，大则更新maxPathSum，情况三
        if (rightInfo != null) {
            maxPathSum = Math.max(maxPathSum, rightInfo.maxPathSum);
        }

        // 去比较左子树连接到x的最大路径和是否是比maxPathSum大，大则更新maxPathSum，情况四
        if (leftInfo != null) {
            // 这个是利用x.val + leftInfo.maxPathSumFromHead求出来的
            maxPathSum = Math.max(maxPathSum, x.val + leftInfo.maxPathSumFromHead);
        }
        // 去比较右子树连接到x的最大路径和是否是比maxPathSum大，大则更新maxPathSum，情况五
        if (rightInfo != null) {
            maxPathSum = Math.max(maxPathSum, x.val + rightInfo.maxPathSumFromHead);
        }
        // 去比较左子树连接到x，再跨国x连接到右子树的最大路径和是否是比maxPathSum大，大则更新maxPathSum，情况六
        if (leftInfo != null && rightInfo != null) {
            // 这个是利用x.val + leftInfo.maxPathSumFromHead + rightInfo.maxPathSumFromHead求出来的
            maxPathSum = Math.max(maxPathSum, x.val + leftInfo.maxPathSumFromHead + rightInfo.maxPathSumFromHead);
        }

        // 返回以当前x节点为根节点的树的Info信息
        return new Info(maxPathSumFromHead, maxPathSum);
    }
}

三、解题思路 

最大路径和一共有如下几种情况：

1、跟x无关

两种可能性

左树、或者右树上的最大路径和

要左树整体的maxPathSum

要右树整体的maxPathSum

然后取左右两棵树maxPathSum的最大值。

2、跟x有关

有四种可能性

1)只有x自己

2)x只往左扎走出来的最大路径和

3)x只往右扎走出来的最大路径和

4)往两头扎出来的最大路径和

通过分析，我们只需要以x为根节点的整棵树的最大路径和（路径并不一定要过x节点） 以及 从左子树或者右子树延伸出来的连到x节点的最大路径和（注意这个只是从左子树或右子树开始连到x为止的路径，并不是从左子树跨过x再连到右子树的路径）。我们只需要这两个信息，就可以组合出上面讲的所有情况。