跳过正文
  1. leetcode 题解/

1022_从根到叶的二进制数之和

·91 字·1 分钟

类型:树

    1. 从根到叶的二进制数之和 💚

    https://leetcode-cn.com/problems/sum-of-root-to-leaf-binary-numbers/

    ❓ 给出一棵二叉树,其上每个结点的值都是 0 或 1 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。例如,如果路径为 0 -> 1 -> 1 -> 0 -> 1,那么它表示二进制数 01101,也就是 13 。

    对树上的每一片叶子,我们都要找出从根到该叶子的路径所表示的数字。

    返回这些数字之和。

    输入:root = [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:
    
            def dfs(node, theSum):
                if not node:
                    return 0
                theSum = (theSum << 1) + node.val
                if not node.left and not node.right:
                    return theSum
                else:
                    return dfs(node.left, theSum) + dfs(node.right, theSum)
    
            return dfs(root, 0)

    时间复杂度:O(n),空间复杂度:O(n)