力扣-124-二叉树中的最大路径和

点我看原题

Java代码

/**
 * 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 {
    static int maxValue;
    public int maxPathSum(TreeNode root) {
        maxValue = Integer.MIN_VALUE;
        dfs(root);
        return maxValue;
    }

    static int dfs(TreeNode root){
        if(root == null) return 0;
        int value = root.val;
        int leftValue = dfs(root.left);
        int rightValue = dfs(root.right);
        leftValue = Math.max(0, leftValue);
        rightValue = Math.max(0, rightValue);
        //我返回的是单条路径最大的值
        //我比较的是我两条路径加起来的最大值
        value += leftValue + rightValue;
        maxValue = maxValue > value ? maxValue : value;
        return value - Math.min(leftValue, rightValue);
    }
}

这一题的关键在于认识到比较用到的maxValue和我们dfs返回的数值并不是同一个
当前节点深搜结束之后，返回的是当前节点的最大子节点和当前节点的值
因为当我们返回到上一层时，一条路径是不能有分叉的，所以我们必须选择一条最大的进行返回
保证后续结果的最大。