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

二叉树#

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

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

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

示例 1：

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

示例 2：

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

递归的解法

作为困难题，其实还是挺简单的

思路：和 543. 二叉树的直径 类似，不过数据处理相对复杂一点，需要处理左右为负的情况，判断最大时需要都加起来，而返回时需要根+左右两边更大的值，这里稍微不一样。

1
/**
2
 * Definition for a binary tree node.
3
 * type TreeNode struct {
4
 *     Val int
5
 *     Left *TreeNode
6
 *     Right *TreeNode
7
 * }
8
 */
9
func diameterOfBinaryTree(root *TreeNode) int {
10
    if root == nil {
11
        return 0
12
    }
13
    maxDiameter := 0
14

15
    var maxDepth func(*TreeNode) int
16
    maxDepth = func(root *TreeNode) int {
17
        if root == nil {
18
            return 0
19
        }
20

21
        leftHeight := maxDepth(root.Left)
22
        rightHeight := maxDepth(root.Right)
23

24
        if leftHeight + rightHeight > maxDiameter {
25
            maxDiameter = leftHeight + rightHeight
26
        }
27

28
        return max(leftHeight, rightHeight) + 1
29
    }
30
    maxDepth(root)
31

32
    return maxDiameter
33
}