《leetcode-go》二叉树根节点到叶子节点和为指定值的路径

给定一个节点数为 n 的二叉树和一个值 sum ,请找出所有的根节点到叶子节点的节点值之和等于的路径,如果没有则返回空。

思路:递归获取路径和即可,用一个栈来存储路径上的节点即可,后进先出。

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

/**
  * 
  * @param root TreeNode类 
  * @param sum int整型 
  * @return int整型二维数组
*/
var pathSumOut [][]int

func pathSum(root *TreeNode, sum int) [][]int {
	// write code here
	if root == nil {
		return pathSumOut
	}
	var path []int
	path = append(path, root.Val)
	getPathSum(root, 0, sum, path)
	return pathSumOut
}
func getPathSum(root *TreeNode, sum int, targartSum int, path []int) {
	if root.Right == nil && root.Left == nil {
		if sum+root.Val == targartSum {
			//需要给path一个全新的存放位置
			pathSumOut = append(pathSumOut, append([]int{}, path...))
		}
		return
	}
	if root.Left != nil {
		path = append(path, root.Left.Val)
		getPathSum(root.Left, sum+root.Val, targartSum, path)
		path = path[:len(path)-1]
	}
	if root.Right != nil {
		path = append(path, root.Right.Val)
		getPathSum(root.Right, sum+root.Val, targartSum, path)
		path = path[:len(path)-1]
	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值