LCR 049. 求根节点到叶节点数字之和(中等)(主站129)

https://leetcode.cn/problems/3Etpl5/
https://leetcode-cn.com/problems/sum-root-to-leaf-numbers/
难度：☆☆☆

题目：

给定一个二叉树的根节点 root ，树中每个节点都存放有一个 0 到 9 之间的数字。每条从根节点到叶节点的路径都代表一个数字：例如，从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。计算从根节点到叶节点生成的 所有数字之和 。
叶节点 是指没有子节点的节点。

示例：

输入：root = [1,2,3]
输出：25
解释：
从根到叶子节点路径 1->2 代表数字 12
从根到叶子节点路径 1->3 代表数字 13
因此，数字总和 = 12 + 13 = 25

解法：DFS

一看这题很容易想到深度优先搜索，从根节点开始遍历，如果遇到叶子节点，则将其加到数字之和。如果不是叶子节点，则计算其子节点对应的数字，然后对子节点递归遍历。
Python

class Solution:
    def sumNumbers(self, root: TreeNode) -> int:
        # node是当前节点，sums是当前节点以上父节点的和
        def dfs(node, sums):
            if not node:
                return 0
            sums = sums * 10 + node.val

            # 如果遇到叶子节点，则将叶子节点对应的数字加到数字之和
            if not node.left and not node.right:
                return sums

            # 如果当前节点不是叶子节点，则计算其子节点对应的数字，然后对子节点递归遍历
            return dfs(node.left, sums) + dfs(node.right, sums)

        return dfs(root, 0)

Java

class Solution {
    public int sumNumbers(TreeNode root) {
        return dfs(root, 0);
    }
    private int dfs(TreeNode node, int sums) {
        if (node == null) {
            return 0;
        }
        sums = sums * 10 + node.val;
        if (node.left == null && node.right == null) {
            return sums;
        }
        return dfs(node.left, sums) + dfs(node.right, sums);
    }
}