sum-root-to-leaf-numbers.go

package problem0129

import (
	"github.com/aQuaYi/LeetCode-in-Go/kit"
)

type TreeNode = kit.TreeNode

func sumNumbers(root *TreeNode) int {
	res := 0

	var dfs func(*TreeNode, int)
	dfs = func(root *TreeNode, temp int) {
		if root == nil {
			return
		}

		temp = temp*10 + root.Val

		if root.Left == nil && root.Right == nil {
			res += temp
			return
		}

		dfs(root.Left, temp)
		dfs(root.Right, temp)
	}

	dfs(root, 0)

	return res
}