否：继续遍历左子树和右子树直到叶节点

得到的路径符合要求时，创建一个新的切片，拷贝当前路径到其中，并添加到 ret 中
• 得到的路径不符合要求，往前回溯，删除当前节点，返回父节点

代码

–执行用时：4 ms --内存消耗：4.5 MB

/**
* Definition for a binary tree node.
* type TreeNode struct {
*     Val int
*     Left *TreeNode
*     Right *TreeNode
* }
*/

func pathSum(root *TreeNode, sum int) [][]int {
if root == nil {
return nil
}
var ret [][]int
dfs(root,sum,[]int{},&ret)
return ret
}

func dfs(root *TreeNode,sum int,arr []int,ret *[][]int){
if root == nil{
return
}
arr = append(arr,root.Val)

if root.Val == sum && root.Left == nil && root.Right == nil {
//slice是一个指向底层的数组的指针结构体
//因为是先序遍历，如果 root.Right != nil ,arr 切片底层的数组会被修改
//所以这里需要 copy arr 到 tmp，再添加进 ret，防止 arr 底层数据修改带来的错误
tmp := make([]int,len(arr))
copy(tmp,arr)
*ret = append(*ret,tmp)
}

dfs(root.Left,sum - root.Val,arr,ret)
dfs(root.Right,sum - root.Val,arr,ret)

arr = arr[:len(arr)-1]
}

在LeetCode该题中，我也有提交题解，欢迎查看。昵称:Sakura。

• 点赞
• 收藏
• 分享
• 文章举报