golang 用栈实现路径求和

题面

输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例: 给定如下二叉树,以及目标和 sum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

返回

[
   [5,4,11,2],
   [5,8,4,5]
]

实现

  • 复用后序遍历逻辑
  • 栈进进出,先进后出,边进边出,出进进
type Stack struct{
     Node *TreeNode
     Sum int
     Path []int
 }
func pathSum(root *TreeNode, sum int) [][]int {
    if root == nil {
        return nil
    }
    rs := [][]int{}
    st :=[]Stack{Stack{Node:root},}
    for len(st) > 0 {
        top := st[len(st)-1]
        st = st[:len(st)-1]
        if top.Node.Left == nil && top.Node.Right == nil && top.Sum + top.Node.Val == sum {
            cp := make([]int,len(top.Path)+1)
            copy(cp, top.Path)
            cp[len(cp)-1]=top.Node.Val
            rs = append(rs, cp)
        }
        if top.Node.Right != nil {
            st = append(st, Stack{
                Node:top.Node.Right,
                Sum: top.Sum + top.Node.Val,
                Path: append(top.Path, top.Node.Val),
            })
        }
        if top.Node.Left != nil {
            st = append(st, Stack{
                Node:top.Node.Left,
                Sum: top.Sum + top.Node.Val,
                Path: append(top.Path, top.Node.Val),
            })
        }
    }
    return rs
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值