题目描述
给定一个仅包含数字\ 0-9 0−9 的二叉树,每一条从根节点到叶子节点的路径都可以用一个数字表示。
例如根节点到叶子节点的一条路径是1\to 2\to 31→2→3,那么这条路径就用\ 123 123 来代替。
找出根节点到叶子节点的所有路径表示的数字之和
例如:
这颗二叉树一共有两条路径,
根节点到叶子节点的路径 1-> 2 用数字 12 代替
根节点到叶子节点的路径 1→3 用数字13 代替
所以答案为 12+13=25
解题思路
- 前序遍历,上层的值乘10,加上当前节点的值,为目前该条路径的结果。
- 如果到叶子节点就返回
- 处理左子树,右子树。将左右子树返回的结果相加返回
代码实现
/*
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
func sumNumbers( root *TreeNode ) int {
sum := 0
return sumHelp(root, sum)
}
func sumHelp(root *TreeNode, sum int) int {
if root == nil {
return 0
}
sum = sum * 10 + root.Val
if root.Left == root.Right {
return sum
}
return sumHelp(root.Left, sum) + sumHelp(root.Right, sum)
}