给出一棵二叉树,其上每个结点的值都是 0 或 1 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。
例如,如果路径为 0 -> 1 -> 1 -> 0 -> 1,那么它表示二进制数 01101,也就是 13 。
对树上的每一片叶子,我们都要找出从根到该叶子的路径所表示的数字。
返回这些数字之和。题目数据保证答案是一个 32 位 整数。
输入:root = [1,0,1,0,1,0,1]
输出:22
解释:(100) + (101) + (110) + (111) = 4 + 5 + 6 + 7 = 22
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/sum-of-root-to-leaf-binary-numbers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
有一个问题要注意的是,如果是个不完全二叉树
# 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: Optional[TreeNode]) -> int:
self.values = []
self.find_tree(root, '')
# print(self.values)
return sum(self.values)
def find_tree(self, root, root_values):
if root.val is None:
return
val = str(root.val)
root_values += val
if root.left or root.right:
if root.left:#先序遍历,根左右
self.find_tree(root.left, root_values)#当最终深度遍历结束,返回到这层
#root_values 还是当时遍历开始的时候的值,子树遍历的值不会影响到当时这一层
if root.right:
self.find_tree(root.right, root_values)
else:
#calculate the binary number to int, 太原始了
# n = len(root_values)
# sum = 0
# for value in root_values:
# sum += int(value) * 2**(n-1)
# n -= 1
sum = int(root_values, 2)#有没有更好的方法,比如位运算?
self.values.append(sum)