用python写leetcode【14】 -- 二叉树中的最大路径和(124)

最新推荐文章于 2024-02-21 14:39:36 发布

原创最新推荐文章于 2024-02-21 14:39:36 发布 · 475 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python

python写leetcode 专栏收录该内容

20 篇文章

订阅专栏

博客围绕二叉树中的最大路径和问题展开，给定非空二叉树，路径为从任意节点出发到任意节点的序列，至少含一个节点且不一定过根节点。介绍了采用dfs方法求解的思路，同时要考虑负数情况，还给出了示例。

二叉树中的最大路径和(124)

题目

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

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

示例 1:

输入: [1,2,3]

   1
  / \
 2   3

输出: 6
示例 2:

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

-10
/
9 20
/
15 7

输出: 42

思路

一般情况下最大的路径是由一个点的左路径和右路径加起来的，因此可以采用dfs的方法。注意要考虑到负数。

代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def maxPathSum(self, root: TreeNode) -> int:   
        self.curr_max = float('-inf')
        def getMax(root):
            if root == None:
                return 0
            left = max(0,getMax(root.left))
            right = max(0,getMax(root.right))
            self.curr_max = max(self.curr_max , left + right + root.val)
            return max(left,right)+root.val
        getMax(root)
        return self.curr_max