给出一棵二叉树,其上每个结点的值都是 0
或 1
。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。例如,如果路径为 0 -> 1 -> 1 -> 0 -> 1
,那么它表示二进制数 01101
,也就是 13
。
对树上的每一片叶子,我们都要找出从根到该叶子的路径所表示的数字。
以 10^9 + 7
为模,返回这些数字之和。
示例:
输入:[1,0,1,0,1,0,1] 输出:22 解释:(100) + (101) + (110) + (111) = 4 + 5 + 6 + 7 = 22
提示:
- 树中的结点数介于
1
和1000
之间。 - node.val 为
0
或1
。
思路:
基础题,DFS+回溯找到每一条从根节点到叶节点的Path,存在res里,
然后线性扫描res这个数组,把每一条path利用int(BIN,2)函数转成十进制,再加到rres中。
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def sumRootToLeaf(self, root):
"""
:type root: TreeNode
:rtype: int
"""
res = []
def dfs(root, tmp):
if not root:
return
tmp += str(root.val)
if not root.left and not root.right:
res.append(tmp[:])
dfs(root.left, tmp)
dfs(root.right, tmp)
tmp = tmp[:-1]
dfs(root, "")
rres = 0
for item in res:
rres += int(item, 2)
rres %= (10 ** 9 + 7)
return rres