这题是求出来一个树中最长的段。大部分的情况都是一个^形状的样子。也就是说你需要找出来左支的最大值+右支的最大值,再加上根就得到了答案。首先采用后序遍历的方式,依次求出每个根的值,然后比较就可以得到最终的结果。
但是这题中出现了负数,这就给题目增加了难度。子节点的值为负数的,对于父节点的值毫无帮助,应该“剪掉”,即,凡是值为负的子节点,一律不参与父节点的计算。这样依次求出所有的节点值,找到最大值即可。
class Solution(object):
def __init__(self):
self.maxofsum = ''
def maxPathSum(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if not root:
return 0
if self.maxofsum == '':
self.maxofsum = root.val
if root.left:
self.maxPathSum(root.left)
if root.right:
self.maxPathSum(root.right)
if not root.left and not root.right:
if self.maxofsum < root.val:
self.maxofsum = root.val
if not root.left and root.right:
root.val += max(root.right.val, 0)
if self.maxofsum < root.val:
self.maxofsum = root.val
if root.left and not root.right:
root.val += max(root.left.val, 0)
if self.maxofsum < root.val:
self.maxofsum = root.val
if root.left and root.right:
if root.val + max(root.left.val, 0) + max(root.right.val, 0) > self.maxofsum:
self.maxofsum = root.val + max(root.left.val, 0) + max(root.right.val, 0)
root.val += max(root.left.val, root.right.val, 0)
return self.maxofsum