求根节点到叶节点数字之和
给你一个二叉树的根节点 root
,树中每个节点都存放有一个 0
到 9
之间的数字。
每条从根节点到叶节点的路径都代表一个数字:
- 例如,从根节点到叶节点的路径
1 -> 2 -> 3
表示数字123
。
计算从根节点到叶节点生成的 所有数字之和 。
输入:root = [1,2,3]
输出:25
解释:
从根到叶子节点路径 1->2 代表数字 12
从根到叶子节点路径 1->3 代表数字 13
因此,数字总和 = 12 + 13 = 25
func sumNumbers(root *TreeNode) int {
var res int = 0
var help func(root *TreeNode, currentSum int)
help = func(root *TreeNode, currentSum int) {
if root == nil {
return
}
if root.Left == nil && root.Right == nil {
res += currentSum*10 + root.Val
}
help(root.Left, currentSum*10+root.Val)
help(root.Right, currentSum*10+root.Val)
}
help(root, 0)
return res
}
叶子节点需要往上路径,所以参数里面带上路径数据。