https://leetcode.com/problems/sum-root-to-leaf-numbers/description/
给定一个二叉树,它的每个结点都存放一个 0-9 的数字,每条从根到叶子节点的路径都代表一个数字。
例如,从根到叶子节点路径 1->2->3 代表数字 123。
计算从根到叶子节点生成的所有数字之和。
说明: 叶子节点是指没有子节点的节点。
示例 1:
输入: [1,2,3]
1
/ \
2 3
输出: 25
解释:
从根到叶子节点路径 1->2 代表数字 12.
从根到叶子节点路径 1->3 代表数字 13.
因此,数字总和 = 12 + 13 = 25.
示例 2:
输入: [4,9,0,5,1]
4
/ \
9 0
/ \
5 1
输出: 1026
解释:
从根到叶子节点路径 4->9->5 代表数字 495.
从根到叶子节点路径 4->9->1 代表数字 491.
从根到叶子节点路径 4->0 代表数字 40.
因此,数字总和 = 495 + 491 + 40 = 1026.
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-root-to-leaf-numbers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
这是一道非常适合训练递归的题目。虽然题目不难,但是要想一次写正确,并且代码要足够优雅却不是很容易。
这里我们的思路是定一个递归的helper函数,用来帮助我们完成递归操作。递归函数的功能是将它的左右子树相加,注意这里不包括这个节点本身,否则会多加, 我们其实关注的就是叶子节点的值,然后通过层层回溯到root,返回即可。
整个过程如图所示:
那么数字具体的计算逻辑,如图所示,相信大家通过这个不难发现规律:
/*
* @lc app=leetcode id=129 lang=javascript
*
* [129] Sum Root to Leaf Numbers
*/
function helper(node, cur) {
if (node === null) return 0;
const next = node.val + cur * 10;
if (node.left === null && node.right === null) return next;
const l = helper(node.left, next);
const r = helper(node.right, next);
return l + r;
}
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @return {number}
*/
var sumNumbers = function(root) {
// tag: `tree` `dfs` `math`
return helper(root, 0);
};
这道题和本题太像了,跟一道题没啥区别