用 Python 编写程序,查找二叉树中任意路径的最大和
pythonserver side programmingprogramming更新于 2026/1/10 11:24:17
假设我们有一棵二叉树,我们必须找到从根节点到叶节点的任意路径的最大和。
因此,如果输入如下

那么输出将是 29,因为从根节点开始,如果我们遵循路径 5-<9-<7-<8,则加法后将是 29。
为了解决这个问题,我们将遵循以下步骤−
定义一个函数 walk()。这将获取节点 s
如果节点为空,则
max_sum := max_sum 和 s 的最大值
返回
s := s + 节点的数据
walk(节点左侧,s)
walk(节点右侧,s)
从主方法执行以下操作−
max_sum := 0
walk(root, 0)
返回 max_sum
让我们看看下面的实现以便更好地理解 −
示例
from collections import defaultdict class TreeNode: def __init__(self, data, left = None, right = None): self.data = data self.left = left self.right = right class Solution: def walk(self, node, s): if not node: self.max_sum = max(self.max_sum, s) return s += node.data self.walk(node.left, s) self.walk(node.right, s) def solve(self, root): self.max_sum = 0 self.walk(root, 0) return self.max_sum ob = Solution() root = TreeNode(5) root.left = TreeNode(1) root.right = TreeNode(9) root.right.left = TreeNode(7) root.right.right = TreeNode(10) root.right.left.left = TreeNode(6) root.right.left.right = TreeNode(8) print(ob.solve(root))
输入
root = TreeNode(5) root.left = TreeNode(1) root.right = TreeNode(9) root.right.left = TreeNode(7) root.right.right = TreeNode(10) root.right.left.left = TreeNode(6) root.right.left.right = TreeNode(8)
输出
29
相关文章
有用资源
python 参考教程 - 该教程包含有关 python 的更多信息:https://www.cainiaomax.com/python/

