LeetCode Algorithm 129. 求根节点到叶节点数字之和

Posted 2022-02-14 Alex_996

129. 求根节点到叶节点数字之和

Ideas

从根节点到叶节点的路径可以通过递归得到，所以我们从根节点开始递归，每次递归的时候还带着一个参数表示从根节点到当前节点所组成的数字对应的字符串，最后如果到了叶节点，累加一下就OK了。

Code

Python

class Solution:
    def sumNumbers(self, root: TreeNode) -> int:
        def dfs(node, string):
            if node.left is None and node.right is None:
                nonlocal ans
                ans += int(string)
                return
            if node.left:
                dfs(node.left, f"stringnode.left.val")
            if node.right:
                dfs(node.right, f"stringnode.right.val")

        ans = 0
        dfs(root, f'root.val')
        return ans