124 二叉树中的最大路径和 困难
发布者:admin 发表于:417天前 阅读数:536 评论:0

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

代码参考:

package main

import "fmt"

func main() {
    root := &TreeNode{Val: 1}
    root.Left = &TreeNode{Val: 2}
    root.Right = &TreeNode{Val: 3}
    // root.Right.Left = &TreeNode{Val: 15}
    // root.Right.Right = &TreeNode{Val: 7}
    fmt.Println(maxPathSum(root))
}

func maxPathSum(root *TreeNode) int {
    if root == nil {
        return 0
    }
    max := -1 << 31
    pathSum(root, &max)
    return max
}

func pathSum(node *TreeNode, maxSum *int) int {
    if node == nil {
        return 0
    }

    lSum := pathSum(node.Left, maxSum)
    rSum := pathSum(node.Right, maxSum)

    curSum := node.Val
    if lSum > 0 {
        curSum += lSum
    }
    if rSum > 0 {
        curSum += rSum
    }

    *maxSum = max(*maxSum, curSum)
    return max(node.Val, node.Val+lSum, node.Val+rSum)
}

func max(nums ...int) int {
    m := nums[0]
    for _, n := range nums[1:] {
        if n > m {
            m = n
        }
    }
    return m
}