题目:
给出一棵二叉树,其上每个结点的值都是 0 或 1 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。例如,如果路径为 0 -> 1 -> 1 -> 0 -> 1,那么它表示二进制数 01101,也就是 13 。
对树上的每一片叶子,我们都要找出从根到该叶子的路径所表示的数字。
返回这些数字之和。题目数据保证答案是一个 32 位 整数。
思路:
先用一个数组res,记录所有路径的数字,最后再转换求和
解答:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def sumRootToLeaf(self, root: TreeNode) -> int:
if not root:
return 0
#用来记录所有的数字
res=[]
q=[[root,'0b']]
while q:
node,tmp=q.pop(0)
tmp+=str(node.val)
if node.left:
q.append([node.left,tmp])
if node.right:
q.append([node.right,tmp])
if not node.left and not node.right:
res.append(tmp)
return sum(int(val,2) for val in res)