Leetcode 124: 二叉树中的最大路径和[GO]

最新推荐文章于 2023-06-24 14:46:07 发布

原创最新推荐文章于 2023-06-24 14:46:07 发布 · 303 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #golang #cpp

Leetcode 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种求解二叉树中最大路径和的算法实现，通过深度优先搜索策略，探讨了四种可能的最大路径情况，即仅根节点、根节点加左子树、根节点加右子树及根节点加左右子树。使用递归函数进行遍历并记录全局最大值，最终返回最大路径和。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

GO

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 * 1. 根节点自己
 * 2. 根节点+左子树
 * 3. 根节点+右子树
 * 4. 根节点+左子树+右子树
 */

import(
    "math"
)

func maxPathSum(root *TreeNode) int {
    maxInt := math.MinInt32
    dfs(root,&maxInt)
    return maxInt
}

func dfs(root *TreeNode,maxInt *int) int {
    if root == nil{
        return 0
    }
    
    left := max(0,dfs(root.Left,maxInt))
    right := max(0,dfs(root.Right,maxInt))
    *maxInt = max(*maxInt,root.Val + left + right)
    return max(left,right) + root.Val
}


func max(a,b int) int{
    if a > b{
        return a
    }
    return b
}