《leetcode-go》二叉树根节点到叶子节点的所有路径和

最新推荐文章于 2024-04-29 11:29:27 发布

原创最新推荐文章于 2024-04-29 11:29:27 发布 · 572 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #golang #算法

leetCode-go 专栏收录该内容

81 篇文章

订阅专栏

这篇博客介绍了一个关于二叉树的问题，目标是计算从根节点到所有叶子节点的数字路径之和。给定的二叉树仅包含0-9的节点，每个路径可以表示一个数字。通过递归函数`sumNumbers`和`sumNum`，遍历树的节点并将路径上的数字累加，最后返回所有路径的总和。

给定一个仅包含数字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
}