二叉树中的最大路径和(Java版)

二叉树中的 路径 被定义为一条节点序列，序列中每对相邻节点之间都存在一条边。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点，且不一定经过根节点。

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

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

解决思路

找最大路径和本质是递归寻找当前节点最大路径

设置全局变量max

        在当前节点左边路径最大值加右边路径最大值大于max，赋值给max

        若小于max，就返回一条路径，左/右路径加当前节点值

        若左/右路径值小于0，那就舍弃，只取当前节点值

所以这个问题解决思路在于找当前节点最大值

        int L = Math.max(0, dfs(root.left));        //左路径若小于0，干脆不要了

        int R = Math.max(0, dfs(root.right));     //右路径同理

        max = Math.max(max, root.val + L + R);         //查询左路径加右路径加当前节点值

 解决

依照上述思路写出以下代码

class Solution {  
    // 定义一个全局变量来存储最大路径和  
    int max = Integer.MIN_VALUE;  
  
    // 主函数，用于启动深度优先搜索并返回最大路径和  
    public int maxPathSum(TreeNode root) {  
        dfs(root); // 从根节点开始深度优先搜索  
        return max; // 返回找到的最大路径和  
    }  
  
    // 深度优先搜索的辅助函数  
    // 该函数返回以当前节点为根的子树中，通过当前节点的最大单边路径和（不包括父节点）  
    int dfs(TreeNode root) {  
        if (root == null) {  
            // 如果当前节点为空，则返回0，表示没有路径和  
            return 0;  
        }  
  
        // 递归地计算左子树和右子树通过各自根节点的最大单边路径和（不包含当前节点）  
        // 注意这里使用Math.max(0, ...)来确保如果子树路径和为负，则忽略该子树  
        int L = Math.max(0, dfs(root.left));  
        int R = Math.max(0, dfs(root.right));  
  
        // 更新全局的最大路径和，考虑当前节点加上左右子树的最大单边路径和  
        max = Math.max(max, root.val + L + R);  
  
        // 返回以当前节点为根的子树中，通过当前节点的最大单边路径和  
        // 这里选择左子树或右子树中较大的一个加上当前节点的值  
        // 注意这里不考虑同时包含左右子树的情况，因为题目要求的是最大单边路径和  
        return Math.max(root.val + L, root.val + R);  
    }   
}