二叉树中的最大路径和,重在递归怎么写。DFS,牢记,只考虑当前代码要干什么,要解决什么问题,然后注意写好return。不要想太多。
不过这个技能也是写了很多递归,思考了很多次才能到这个境界的,初学者还是多写多想。但是最终目标就是,去解决当前问题,这个境界。这样DFS就不会写错。
好了,题目:
124. 二叉树中的最大路径和
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
输入: [1,2,3]
1
/ \
2 3
输出: 6
示例 2:
输入: [-10,9,20,null,null,15,7]
-10
/ \
9 20
/ \
15 7
输出: 42
解答:
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def maxPathSum(self, root: TreeNode) -> int:
self.maxnum = -float('inf')
def dfs(root):
if not root:
return 0
left = dfs(root.left)
right = dfs(root.right)
self.maxnum = max(self.maxnum, left+root.val+right, left+root.val, right+root.val, root.val)
return max(root.val+left, root.val+right, root.val)
dfs(root)
return self.maxnum
好了。