牛客题霸 NC9 二叉树中是否存在节点和为指定值的路径

Posted 2021-09-04 Starzkg

https://www.nowcoder.com/practice/508378c0823c423baa723ce448cbfd0c

解决方案

Go

func hasPathSum(root *TreeNode, sum int) bool {
	// write code here
	return dfs(root, sum)
}

func dfs(root *TreeNode, sum int) bool {
	if root == nil {
		return false
	}
	sum = sum - root.Val
	if root.Left == nil && root.Right == nil {
		if sum == 0 {
			return true
		}
		return false
	}
	return dfs(root.Left, sum) || dfs(root.Right, sum)
}

参考文章

牛客题霸 NC8 二叉树根节点到叶子节点和为指定值的路径