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

package main

import . "nc_tools"

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

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 *
 * @param root TreeNode类
 * @param sum int整型
 * @return int整型
 */
func FindPath(root *TreeNode, sum int) int {
	if root == nil {
		return 0
	}
	tempCount := 0
	if sum-root.Val == 0 {
		tempCount= tempCount+1
	}
    if root.Left!=nil{
	   tempCount = tempCount+ FindPath(root.Left, sum)
       tempCount = tempCount+ dfs(root.Left, sum-root.Val) 
	}
	if root.Right!=nil{
		tempCount = tempCount+ FindPath(root.Right, sum)
		tempCount = tempCount+ dfs(root.Right, sum-root.Val) 
	}
	return tempCount
}


func dfs(root *TreeNode, sum int) int{
	if root == nil {
		return 0
	}
	tempCount := 0
	if sum== root.Val {
		tempCount= tempCount+1
	}
    if root.Left!=nil{
       tempCount = tempCount+ dfs(root.Left, sum-root.Val) 
	}
	if root.Right!=nil{
		tempCount = tempCount+ dfs(root.Right, sum-root.Val) 
	}
	return tempCount
}


需要注意从每一个子节点开始的时候，总数需要不变，总数变的话直接走子函数就行了