124. 二叉树中的最大路径和
-
解题思路:二叉树递归套路,以某个节点
node
为根,最大路径是什么?- 1️⃣与
node
无关,也就是说,不经过node
- 左子树的最大值,与右子树的最大值,较大的那个,就是结果
- 2️⃣与
node
有关- 单独
node
一个节点 - 经过左子树
- 经过右子树
- 左右子树都经过
- 单独
- 所以我们需要的信息就两个:
total_max
整棵树上的最大值,为了得到1️⃣,single_max
,单边的最大值,为了得到2️⃣
- 1️⃣与
-
代码
class Solution: class Info: def __init__(self, total_max, single_max): self.total_max = total_max # 整棵树的最大值 self.single_max = single_max # 单边的最大值(只能走左子树,或者走右子树,不能同时走,并且包含该节点的值) def process(self, node: Optional[TreeNode]) -> Optional[Info]: if node == None: return self.Info(0, 0) left_info = self.process(node.left) right_info = self.process(node.right) # 单边最大值:1)左边单边最大值+val 2)右边单边最大值+val 3)val single_max = node.val + max(0, left_info.single_max, right_info.single_max) # 整体最大值:1)和本节点无关:左子树最大值 2)和本节点无关:右节点最大值 # 3)单独本节点:val 4)本节点+左单边 5)本节点+右单边 total_max = node.val + max(0, left_info.single_max) + max(0, right_info.single_max) if node.left: total_max = max(left_info.total_max, total_max) if node.right: total_max = max(right_info.total_max, total_max) return self.Info(total_max, single_max) def maxPathSum(self, root: Optional[TreeNode]) -> int: Info = self.process(root) return Info.total_max
标签:
leetcode刷题
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理