给定一个仅包含数字0-9 的二叉树,每一条从根节点到叶子节点的路径都可以用一个数字表示。
例如根节点到叶子节点的一条路径是,那么这条路径就用 来代替。
找出根节点到叶子节点的所有路径表示的数字之和
package main
import . "nc_tools"
/*
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
/**
*
* @param root TreeNode类
* @return int整型
*/
func sumNumbers(root *TreeNode) int {
// write code here
if root == nil {
return 0
}
return sumNum(root, 0)
}
func sumNum(node *TreeNode, sum int) int {
sum = sum*10 + node.Val
if node.Left == nil && node.Right == nil {
return sum
}
var left = 0
if node.Left != nil {
left = sumNum(node.Left, sum)
}
var right = 0
if node.Right != nil {
right = sumNum(node.Right, sum)
}
return left + right
}