leetcode-129-求根节点到叶节点数字之和

943 阅读2分钟

我正在参加「掘金·启航计划」
题目地址

给你一个二叉树的根节点 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

解题思路

本题要求我们计算根节点到所有叶节点的路径之和,所以我们只需要一个前序遍历,在遍历过程中记录当前路径的节点之和,当遇到叶节点时,说明当前路径已经走到头,将和值累加到结果值即可。这样当整棵树处理完成,结果值就保存了我们需要的根节点到所有叶节点的路径之和。

代码实现

var sumNumbers = function(root) {
    // 初始化结果
    let total = 0

    function getNum(node,sum){
        // 累加 sum
        sum = sum*10+node.val
        // 如果当前为叶节点,累加结果值
        if(!node.left && !node.right){
            total += sum
        }
        // 否则继续向下处理子节点
        if(node.left){
            getNum(node.left,sum)
        }
        if(node.right){
            getNum(node.right,sum)
        }
    }

    getNum(root,0)
    return total
};

至此我们就完成了 leetcode-129-求根节点到叶节点数字之和

如有任何问题或建议,欢迎留言讨论!