打印树的和的路径

题目

给定一个二叉树,找到从根节点到叶子节点的和与给定目标相等的路径

思路

从根节点开始,遍历每一条路径,每走过一个节点,就把目标值减去这个节点的值,当最后一个节点没有左右孩子且这个节点的值等于剩余的目标值,就返回这个路径,用一个局部list存当前路径。

  def sum(self,target):
    return self.sum_helper(self.root,target)

  def sum_helper(self,node,target):
    if node is None:
      return False
    if node._left is None and node._right is None and node._item == target:
    #if node._item == target:
      return True
    target = target - node._item

    return self.sum_helper(node._left,target) or self.sum_helper(node._right,target)
  

  def sum_road(self,target):
    return self.sum_target_helper(self.root,target)
  def sum_target_helper(self,node,target):
    if node is None:
      return []
    res = []
    self.dfs(node,target,[],res)
    return res
  def dfs(self,node,target,ls,res):
    if node._left is None and node._right is None and node._item == target:
      ls.append(node._item)
      res.append(ls)
    if node._left:
      self.dfs(node._left,target-node._item,ls+[node._item],res)
    if node._right:
      self.dfs(node._right,target-node._item,ls+[node._item],res)
def sum(self,target):
    return self.sum_helper(self.root,target)

  def sum_helper(self,node,target):
    if node is None:
      return False
    #if node._left is None and node._right is None and node._item == target:
    if node._item == target:
      return True
    target = target - node._item

    return self.sum_helper(node._left,target) or self.sum_helper(node._right,target)
  

  def sum_road(self,target):
    return self.sum_road_all(self.root,target)
    #return self.sum_target_helper(self.root,target)
  def sum_road_all(self,node,target):
    if node is not None:
      return self.sum_target_helper(node,target)+self.sum_target_helper(node._left,target)+self.sum_target_helper(node._right,target)
  def sum_target_helper(self,node,target):
    if node is None:
      return []
    res = []
    self.dfs(node,target,[],res)
    return res
  def dfs(self,node,target,ls,res):
   # if node._left is None and node._right is None and node._item == target:
    if node._item == target:
      ls.append(node._item)
      res.append(ls)
    if node._left:
      self.dfs(node._left,target-node._item,ls+[node._item],res)
    if node._right:
      self.dfs(node._right,target-node._item,ls+[node._item],res)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值