【路飞】算法与数据结构-二叉树中的最大路径和

126 阅读1分钟

不管全世界所有人怎么说,我都认为自己的感受才是正确的。无论别人怎么看,我绝不打乱自己的节奏。喜欢的事自然可以坚持,不喜欢的怎么也长久不了。

LeetCode:原题地址

题目要求

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

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

给你一个二叉树的根节点 root ,返回其 最大路径和 。

示例 1:

exx1.jpg

输入: root = [1,2,3]
输出: 6
解释: 最优路径是 2 -> 1 -> 3 ,路径和为 2 + 1 + 3 = 6

示例 2:

exx2.jpg

输入:root = [-10,9,20,null,null,15,7]
输出:42
解释:最优路径是 15 -> 20 -> 7 ,路径和为 15 + 20 + 7 = 42

提示:

  • 树中节点数目范围是 [1, 3 * 104]
  • -1000 <= Node.val <= 1000

思路

  • 可以先遍历一次树,计算从每个节点出发单条路径节点最大值。以 [-10,9,20,null,null,15,7] 为例。第一次遍历完可以得到从每个节点出发单条路径下的最大值。

  • 再遍历一次树,利用上一次计算的结果,计算以每个节点为根,它能获得的最大路径和(单条路径最大值 + 较小的子节点的最大值)。

/**
 * @param {TreeNode} root
 * @return {number}
 */
var maxPathSum = function(root) {
    if (!root) {
        return Number.MIN_SAFE_INTEGER
    }
    function dfs1 (node) {
        if (!node) {
            return 0
        }
        const L = dfs1(node.left)
        const R = dfs1(node.right)
        node.val = Math.max(L, R, 0) + node.val
        return node.val
    }
    dfs1(root)

    let ans = Number.MIN_SAFE_INTEGER
    function dfs2 (node) {
        if (!node) {
            return Number.MIN_SAFE_INTEGER
        }
        ans = Math.max(ans, node.val + Math.max(Math.min(node.left?.val || 0, node.right?.val || 0), 0))
        dfs2(node.left)
        dfs2(node.right)
    }
    dfs2(root)
    return ans
};