牛客题霸 NC5 二叉树根节点到叶子节点的所有路径和

Posted 2021-09-04 Starzkg

https://www.nowcoder.com/practice/185a87cd29eb42049132aed873273e83

解决方案

Go

func sumNumbers(root *TreeNode) int {
	// write code here
	return sum(0, root)
}

func sum(pre int, root *TreeNode) int {
	if root == nil {
		return 0
	}
	res := pre*10 + root.Val
	if root.Left != nil || root.Right != nil {
		return sum(res, root.Left) + sum(res, root.Right)
	}
	return res
}

参考文章