?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

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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容