根到叶子结点的数字之和

package main

import "fmt"

type TreeNode struct {
    Val   int
    Left  *TreeNode
    Right *TreeNode
}

func sumNumbers(root *TreeNode) int {
    return dfs(root, 0)
}

func dfs(node *TreeNode, currentSum int) int {
    if node == nil {
        return 0
    }

    // 计算当前路径上的数字之和
    currentSum = currentSum*10 + node.Val

    // 如果是叶子结点，返回当前路径的数字之和
    if node.Left == nil && node.Right == nil {
        return currentSum
    }

    // 递归计算左右子树的数字之和
    return dfs(node.Left, currentSum) + dfs(node.Right, currentSum)
}

func main() {
    // 创建一个二叉树
    root := &TreeNode{Val: 1}
    root.Left = &TreeNode{Val: 2}
    root.Right = &TreeNode{Val: 3}
    root.Left.Left = &TreeNode{Val: 4}
    root.Left.Right = &TreeNode{Val: 5}

    // 计算根到叶子结点的数字之和
    result := sumNumbers(root)

    // 输出结果
    fmt.Printf("根到叶子结点的数字之和是：%d\n", result)
}