题解 | #二叉树中和为某一值的路径(三)#

package main


import . "nc_tools"

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

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 *
 * @param root TreeNode类
 * @param sum int整型
 * @return int整型
 */
var ans = 0

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

	if root.Val == sum {
		ans++
	}

	dfs(root.Left, sum-root.Val)
	dfs(root.Right, sum-root.Val)
}

func FindPath(root *TreeNode, sum int) int {

	if root == nil {
		return ans
	}

	dfs(root, sum)

	FindPath(root.Left, sum)
	FindPath(root.Right, sum)
	return ans
}