更多精彩内容,请关注【力扣简单题】。
题目
难度:★★☆☆☆
类型:字符串
给出一棵二叉树,其上每个结点的值都是 0 或 1 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。例如,如果路径为 0 -> 1 -> 1 -> 0 -> 1,那么它表示二进制数 01101,也就是 13 。
对树上的每一片叶子,我们都要找出从根到该叶子的路径所表示的数字。
以 10^9 + 7 为模,返回这些数字之和。
提示
树中的结点数介于 1 和 1000 之间。
node.val 为 0 或 1 。
示例
输入:[1,0,1,0,1,0,1]
输出:22
解释:(100) + (101) + (110) + (111) = 4 + 5 + 6 + 7 = 22
解答
class Solution:
def sumRootToLeaf(self, root: TreeNode) -> int:
record = []
sm = 0
result = 0
# 使用中序遍历来求和
while record or root:
while root:
sm = 2*sm + root.val
record.append([root, sm])
root = root.left
# 如果是已经遍历到最左,开始通过pop来向root方向返回遍历时
# 就不会执行上面的while,而直接pop,重新给sm赋值
# 所以record里存sm是有用的
node, sm = record.pop()
# 只有到叶子节点才将结果累加到result
if not node.left and not node.right:
result += sm
root = node.right
return result
如有疑问或建议,欢迎评论区留言~