剑指Offer|LCR 049. 求根节点到叶节点数字之和

LCR 049. 求根节点到叶节点数字之和

给定一个二叉树的根节点 root ，树中每个节点都存放有一个 0 到 9 之间的数字。

每条从根节点到叶节点的路径都代表一个数字：

• 例如，从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。

计算从根节点到叶节点生成的 所有数字之和 。

叶节点 是指没有子节点的节点。

示例 1：

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

示例 2：

输入：root = [4,9,0,5,1]
输出：1026
解释：
从根到叶子节点路径 4->9->5 代表数字 495
从根到叶子节点路径 4->9->1 代表数字 491
从根到叶子节点路径 4->0 代表数字 40
因此，数字总和 = 495 + 491 + 40 = 1026

提示：

• 树中节点的数目在范围 [1, 1000] 内
• 0 <= Node.val <= 9
• 树的深度不超过 10

法1：递归

分析：

1. 定义终止条件：当root为空，则直接返回该0；当没有左右孩子，。
2. 分解问题：不断遍历root的左孩子以及右孩子

function TreeNode(val, left, right) {
    this.val = (val===undefined ? 0 : val)
    this.left = (left===undefined ? null : left)
    this.right = (right===undefined ? null : right)
}

var sumNumbers = function(root) {
    return dfs(root, 0);
};

function dfs(root, path){
    // 结点为空，则代表数字0
    if(root === null) return 0;

    path = path * 10 + root.val; // 数字值
    if(root.left === null && root.right === null){
        return path;
    }

    return dfs(root.left, path) + dfs(root.right, path);
}

let root = {
    val: 1,
    left: { val: 2, left: null, right: null },
    right: { val: 3, left: null, right: null }
};
console.log(sumNumbers(root));  // 输出：25