【leetcode 1022. 从根到叶的二进制数之和】Python 解题思路

给出一棵二叉树，其上每个结点的值都是 0 或 1 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。例如，如果路径为 0 -> 1 -> 1 -> 0 -> 1，那么它表示二进制数 01101，也就是 13 。
对树上的每一片叶子，我们都要找出从根到该叶子的路径所表示的数字。
以 10^9 + 7 为模，返回这些数字之和。

题目链接：
开心，又实践了一下nonlocal的用法，对于二进制数的操作还是得锻炼熟悉，对于二叉树的递归操作还得加深理解，不然用的时候什么也想不起来
以下两句等价

base = 2 * base + root.val
base = base<<2 | root.val

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def sumRootToLeaf(self, root: TreeNode) -> int:
        ans=0
        
        def helper(root, base):
            nonlocal ans
            if root == None:
                return 0
            if root.left == None and root.right == None:
                ans+=2 * base + root.val
            base = 2 * base + root.val
            helper(root.left, base)
            helper(root.right, base)
        helper(root,0)
        return ans