LeetCode0124.二叉树中的最大路径和 Go语言AC笔记

 时间复杂度:O(n)

解题思路

        对二叉树进行后序遍历。

        为什么选择后序遍历呢?因为后序遍历能先遍历孩子结点,这样就能根据孩子结点的最大值来更新根节点的最大值。

        在遍历时,需要记录的是根节点与一个孩子组成的路径和和根节点与两个组成的路径和。

        对于前者的意义是为了返回给该结点的父节点,尝试让父节点与该结点组成一条路径。根节点与一个孩子组成的路径和要与根节点自身的大小作比较取较大值,因为可能孩子都是负值,选择与一个孩子组成路径会造成路径和减少的情况发生。

        对于后者的意义是为了获得题目要求的最大路径和。如果一个结点的两个孩子都是正数,那么该结点组成的左孩子→根节点→右孩子路径是保证有最大值的。不过也有可能一个孩子是负数,所以我们要与刚才求的根节点与一个孩子组成的路径和做比较,取较大值。

AC代码

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func maxPathSum(root *TreeNode) int {
    maxSum:=math.MinInt32
    var getMax func(root *TreeNode)int
    getMax=func(root *TreeNode)int{
        if root==nil{
            return 0
        }
        leftMax:=getMax(root.Left)
        rightMax:=getMax(root.Right)
        pathMax:=max(root.Val+max(leftMax,rightMax),root.Val)//根节点和一个孩子组成的路径
        newMaxSum:=max(root.Val+leftMax+rightMax,pathMax)//根节点和左右两个孩子组成的一条路径,可能是最大路径和
        maxSum=max(maxSum,newMaxSum)//更新最大路径he
        return pathMax//返回的是根节点和一个孩子组成的路径长度,因为该结点的父节点可能会与该结点组成一条路径
    }
    getMax(root)
    return maxSum
}
func max(a,b int)int{
    if a>b{
        return a
    }
    return b
}

感悟

这题说简单也简单,说难也难,主要还是看对递归熟不熟悉,已经想好怎样能获得到最大路径和。

后序遍历,求左右孩子的最大路径和,然后看是否超过自己,更新自己和所有孩子的路径和,最后返回自己或者自己和一个孩子的路径和。

你可能感兴趣的:(LeetCode,#,树,golang,算法,力扣,数据结构)