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

描述

给定一个非空二叉树，返回其最大路径和。

本题中，路径被定义为一条从树中任意节点出发，达到任意节点的序列。该路径至少包含一个节点，且不一定经过根节点。

示例 1:

输入: [1,2,3]

       1
      / \
     2   3

输出: 6
示例2:

输入: [-10,9,20,null,null,15,7]

   -10
   / \
  9  20
    /  \
   15   7

输出: 42

思路

注意 
1. 递归的时候 子树放回的值 不一定是子树的最大路径和  需要指定一个变量 将子树的最大路径和保存下来
2. 递归返回的时候 必须要是左/右 子树 加上根节点 或者只有根节点 不然这个子树返回时路径就断了

实现

// 初版
var IntMax = int(^uint(0) >> 1)
func maxPathSum(root *TreeNode) int {
    if root == nil{
        return 0
    }

    var max = ^IntMax
    maxPathSumHelp(root, &max)

    return max
}

// 本函数获取的是节点可提供的最大路径和并不是本节点的最大路径和 而是节点的半个分支 ：即左分支和右分支和本节点的最大值
func maxPathSumHelp(root *TreeNode, max *int)int{
    var left int
    var right int
    var rootMax int		// 统计本节点的最大路径和
    
	// 获取左子节点可提供的最大路径和
    if root.Left != nil{
        left = maxPathSumHelp(root.Left, max)
    }

	// 获取右子节点可提供的最大路径
    if root.Right != nil{
        right = maxPathSumHelp(root.Right, max)
    }

	// 如果大于0 
    if left > 0{
    	// 本节点的最大路径和+left
        rootMax = rootMax + left
    }

    if right > 0{
    	// 本节点的最大路径和+left
        rootMax = rootMax + right
    }
	
	rootMax = rootMax + root.Val	// rootMax 则为通过本节点的最大路径和
    *max = Max(*max, rootMax)		// 将rootMax 和 max的较大值赋值给max

	// 返回本节点可提供给父节点的最大路径 如果left和right都小于0 则值返回root.Val 否则返回较大的分支+root.Val
    if Max(left, right) > 0{
        return Max(left, right) + root.Val
    }else{
        return root.Val
    }
}

// 第二版
func Max(a, b int) int {
	if a >= b {
		return a
	} else {
		return b
	}
}

func MaxPathSum(root *TreeNode) int {
	var intMax = int(^uint(0) >> 1)
	var intMin = ^intMax

	var max = intMin // max 初始值为int 的最小值， 因为在递归时 有可能会有零值

	DoMaxPathSum(root, &max)

	return max
}

func DoMaxPathSum(root *TreeNode, max *int) int{
	var maxL int
	var maxR int
	var maxRoot int

	if root == nil{
		return 0
	}

	maxL = Max(DoMaxPathSum(root.Left, max), 0)  // 子树如果为负值 则没有必要经过了

	maxR = Max(DoMaxPathSum(root.Right, max), 0)

	maxRoot = maxL + maxR + root.Val  // 当前节点能贡献的最大值

	*max = Max(*max, maxRoot)

	return root.Val + Max(maxL, maxR) // 必须经过根节点 否则路径会断
}