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

最新推荐文章于 2020-09-27 10:35:27 发布

洛洛洛洛洛洛洛

最新推荐文章于 2020-09-27 10:35:27 发布

阅读量398

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/weixin_44740082/article/details/108350663

leetcode 专栏收录该内容

242 篇文章

订阅专栏

本文详细解析了一道经典的二叉树算法题——寻找所有和为目标值的路径。通过递归和非递归两种方法，阐述了如何从根节点到叶节点找到所有节点值之和等于给定整数的路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接

题目描述：

输入一棵二叉树和一个整数，打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

示例:
给定如下二叉树，以及目标和 sum = 22，

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

[
[5,4,11,2],
[5,8,4,5]
]
解题思路：
遍历每一个节点，记录节点的值，并将该节点添加到路径中，当路径总和等于sum时，将该路径添加到返回结果中
在返回每个节点的父节点时，将该节点弹出路径
p.s. 记录路径时若直接执行 res.append(path) ，则是将 path 列表对象加入了 res ；后续 path 对象改变时， res 中的 path 对象也会随之改变（因此肯定是不对的，本来存的是正确的路径 path ，后面又 append 又 pop 的，就破坏了这个正确路径）。list(path) 相当于新建并复制了一个 path 列表，因此不会受到 path 变化的影响。
p.s.2 tmp,res可以放在helper前面定义，这样就不用在参数中传递

class Solution:
    def pathSum(self, root: TreeNode, sum: int) -> List[List[int]]:
        
        def helper(root,tar,tmp,res):
            if not root:
                return
            tar-=root.val
            tmp.append(root.val)
            if tar==0 and  not root.left  and  not root.right:#是叶子节点并且符合条件
                res.append(list(tmp))
            helper(root.left,tar,tmp,res)
            helper(root.right,tar,tmp,res)
            tmp.pop() 
        tmp,res=[],[]
        helper(root,sum,tmp,res)
        return res

评论区答案，非递归做法

def pathSum(self, root: TreeNode, target: int) -> List[List[int]]:
    if not root:return []
    stack=[(root,[root.val])]
    res=[]
    while stack:
        node,val=stack.pop()
        if node and not node.left and not node.right and sum(val)==target:
            res.append(val)
        if node.left:
            stack.append((node.left,val+[node.left.val]))
        if node.right:
            stack.append((node.right,val+[node.right.val]))
    return res