124. 二叉树中的最大路径和

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

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

示例 1:

输入: [1,2,3]

   1
  / \
 2   3

输出: 6

class Solution(object):
    def maxPathSum(self, root):
        #需要一个函数去求单支路上的最大值
        self.res = float('-inf') #记录最大路径和
        self.max1(root) #在求单支路最大值的过程中记录最大路径和
        return self.res
    def max1(self, root):
        if not root:
            return 0
        left = max(self.max1(root.left), 0)#左支路最大路径
        right = max(self.max1(root.right), 0)#右支路
        self.res = max(self.res, root.val+left+right)
        tmp = max(left, right) + root.val
        return tmp if tmp>0 else 0

你可能感兴趣的:(刷题)