golang刷leetcode 二叉树的最大路径和

2022-08-02 15:46:32 浏览数 (1)

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

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

示例 1:

代码语言:javascript复制
输入: [1,2,3]

       1
      / 
     2   3

输出: 6

示例 2:

代码语言:javascript复制
输入: [-10,9,20,null,null,15,7]

   -10
   / 
  9  20
    /  
   15   7

输出: 42

解题思路:

对于二叉树问题优先想到递归,因为划分子问题比较容易,最大路径和隐含问题是路径连续

1,由于可能含根,可能不含根,所以最大和为

max(根的值,左分支含根最大和,左分支含根最大和+根,右分支含根最大和,右分支含根最大和+根,左分支含根最大和+根+右分支含根最大和,左分支不含根最大和,右分支不含根最大和)

2,上述问题包含含根(单边)最大和子问题,求解为

max(根的值,根的值+左含根最大和,根的值+右含根最大和)

注意不包含:左含根最大和+根的值+右含根最大和,因为路线不能有分叉

代码如下

代码语言:javascript复制
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func maxPathSum(root *TreeNode) int {
    if root==nil{
        return 0
    }
    if root.Left!=nil && root.Right!=nil{
    lr:=maxPathSumRoot(root.Left)
    rr:=maxPathSumRoot(root.Right)
    l:=maxPathSum(root.Left)
    r:=maxPathSum(root.Right)
    a:=[]int{root.Val,root.Val lr,root.Val rr,lr,rr,root.Val lr rr,l,r}
         return max(a)
    }else if root.Left!=nil {
         lr:=maxPathSumRoot(root.Left)
          l:=maxPathSum(root.Left)
         a:=[]int{root.Val,root.Val lr,lr,l}
         return max(a)
    }else if root.Right!=nil{
          rr:=maxPathSumRoot(root.Right)
         r:=maxPathSum(root.Right)
         a:=[]int{root.Val,root.Val rr,rr,r}
         return max(a)
    }
        return root.Val

}

func max(arr []int)  int{
    max:=-1<<31
    for i:=0;i<len(arr);i  {
        if max<arr[i]{
            max=arr[i]
        }
    }
    return max
}

func maxPathSumRoot(root *TreeNode) int {
    if root==nil{
        return 0
    }   
    l:=maxPathSumRoot(root.Left)
    r:=maxPathSumRoot(root.Right)
    return max([]int{l root.Val,r root.Val,root.Val}) 
}

0 人点赞