leetcode 124. 二叉树中的最大路径和

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

思路

这题实际上后序遍历比较方便，先把两个子树最大的路径和返上来。然后决定当前这个节点为根节点的树能达成的最大路径和是多少，然后再把这个节点作为最长路径的一部分的结果返上去，有点dp 那味

题解

/**
 * 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 {
    int res = -100000;
    public int maxPathSum(TreeNode root) {
        dfs(root);
        return res;
    }

    public int dfs(TreeNode root) {
        if (root == null) {
            return 0;
        }
        // 不管左子树和右子树是咋来的值，而且只有正的才有意义
        int left = Math.max(0, dfs(root.left));
        int right = Math.max(0, dfs(root.right));
        // 每到一个节点就算一下以此节点为根的树，路径最大值
        res = Math.max(res, left + right + root.val);
        // 返上去的话肯定只能返一边子树加上根节点
        // 不能不带根节点，要不连不上去上面
        return Math.max(left + root.val, right + root.val);
    }
}