剑指 Offer 34. 二叉树中和为某一值的路径

题目描述

给你二叉树的根节点 root 和一个整数目标和 targetSum ，找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。

叶子节点 是指没有子节点的节点。

示例 1：

输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出：[[5,4,11,2],[5,8,4,5]]
示例 2：

输入：root = [1,2,3], targetSum = 5
输出：[]
示例 3：

输入：root = [1,2], targetSum = 0
输出：[]

提示：

树中节点总数在范围 [0, 5000] 内
-1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000

解答思路

1. 使用匿名函数处理单次的根节点到叶子的数据
2. 在回溯的时候注意当前节点数据的清除

优化记录

无

提交结果

package main

import "fmt"

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

func pathSum(root *TreeNode, target int) (ams [][]int) {
    resultRes := []int{}
    //注意此处是切片，切片和数组的区别是初始化的时候是否限制长度
    var dfs func(node *TreeNode, leftValue int)
    dfs = func(node *TreeNode, leftValue int) {
        if node == nil {
            return
        }
        leftValue -= node.Val
        resultRes = append(resultRes, node.Val)
        defer func() { resultRes = resultRes[:len(resultRes)-1] }()
        //是否需要在 defer 语句或赋值语句后面使用 () 取决于匿名函数是否需要在执行时传递参数或调用函数。如果需要传递参数或调用函数，则需要使用 () 来表示函数调用；如果匿名函数没有参数，可以省略 ()。
        if node.Left == nil && node.Right == nil && leftValue == 0 {
            ams = append(ams, append([]int(nil), resultRes...))
            //使用了两次append，append([]int(nil), resultRes...)表示将resultRes存储为新的切片，后续切片操作不影响当前操作
            return
        }
        dfs(node.Left, leftValue)
        dfs(node.Right, leftValue)
    }
    dfs(root, target)
    return
}

func main() {
    // 创建一个测试二叉树
    root := &TreeNode{
        Val: 5,
        Left: &TreeNode{
            Val: 4,
            Left: &TreeNode{
                Val: 11,
                Left: &TreeNode{
                    Val: 7,
                },
                Right: &TreeNode{
                    Val: 2,
                },
            },
        },
        Right: &TreeNode{
            Val: 8,
            Left: &TreeNode{
                Val: 13,
            },
            Right: &TreeNode{
                Val: 4,
                Left: &TreeNode{
                    Val: 5,
                },
                Right: &TreeNode{
                    Val: 1,
                },
            },
        },
    }

    target := 22

    // 调用 pathSum 函数
    result := pathSum(root, target)

    // 打印结果
    fmt.Println(result)
}

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。