- Notifications
You must be signed in to change notification settings - Fork 20
/
Copy path124.BinaryTreeMaximumPathSum_zheyuuu.py
51 lines (48 loc) · 1.28 KB
/
124.BinaryTreeMaximumPathSum_zheyuuu.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
# Source : https://leetcode.com/problems/binary-tree-maximum-path-sum
# Author : zheyuuu
# Date : 2020-08-05
#####################################################################################################
#
# Given a non-empty binary tree, find the maximum path sum.
#
# For this problem, a path is defined as any sequence of nodes from some starting node to any node in
# the tree along the parent-child connections. The path must contain at least one node and does not
# need to go through the root.
#
# Example 1:
#
# Input: [1,2,3]
#
# 1
# / \
# 2 3
#
# Output: 6
#
# Example 2:
#
# Input: [-10,9,20,null,null,15,7]
#
# -10
# / \
# 9 20
# / \
# 15 7
#
# Output: 42
#
#####################################################################################################
classSolution:
defmaxPathSum(self, root: TreeNode) ->int:
self.ans=float("-inf")
self.helper(root)
returnself.ans
defhelper(self, node):
ifnotnode:
return0
left=self.helper(node.left)
right=self.helper(node.right)
lval=leftifleft>=0else0
rval=rightifright>=0else0
self.ans=max(node.val+lval+rval, self.ans)
returnmax(node.val+lval, node.val+rval)