？1022. 从根到叶的二进制之和（Python）

更多精彩内容，请关注【力扣简单题】。

题目

难度：★★☆☆☆
类型：字符串

给出一棵二叉树，其上每个结点的值都是 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

如有疑问或建议，欢迎评论区留言~