【力扣】1022:从根到叶的二进制数之和 | DFS

题目描述

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

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

以 10^9 + 7 为模,返回这些数字之和。

算法思路

其实就是遍历所有从根到叶的路径。换汤不换药的题。确实简单。

class Solution:
    res=0
    def sumRootToLeaf(self, root: TreeNode,pre='') -> int:
        if root:
            if not root.left and not root.right:
                self.res+=int(pre+str(root.val),2)
                return self.res
            self.sumRootToLeaf(root.left,pre+str(root.val))
            self.sumRootToLeaf(root.right,pre+str(root.val))
            return self.res

你可能感兴趣的:(力扣日记)