Leetcode刷题:剑指offer【面试题34 二叉树中和为某一值的路径】

文章目录

【面试题34 二叉树中和为某一值的路径】

难度: 中等

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

Leetcode题目对应位置: 面试题34:二叉树中和为某一值的路径

思路:回溯法,正常操作就是先序遍历二叉树,并记录路径和。按照 “根左右” 顺序遍历二叉树,同时记录从根节点到当前节点的路径及路径和,当路径同时满足:1)是从根节点到叶节点的路径,2)各节点值之和等于目标值时,将该路径加入列表。

代码逻辑

  • 添加新路径节点:遍历到当前节点 root,将其值 root.val 加入路径 path
  • 更新当前路径和:遍历到当前节点 root,将其值加到当前路径的总和之中
  • 检查是否满足要求:检查是否满足 1)当前节点为叶节点,且 2)当前路径之和等于目标值,若满足则将路径 path 加入结果列表 res
  • 向下递归:递归左/右子节点
  • 路径回溯:若当前节点导致路径和大于目标值,向上回溯,将最新进入的节点从 path 中删除,且路径和减去其值。

时间复杂度:O(n),需要遍历节点个数为 n 的二叉树
空间复杂度:O(n),最差情况下,二叉树退化为链表,递归调用栈需要 O(n)

注意:向结果列表中添加 path 列表时,如果直接 res.append(path),相当于将 path 对象加入了 res,后面 path 如果改变,res 中的 path 也会改变。正确做法是 res.append(list(path)),相当于复制一个 path 加入到 res 中。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def pathSum(self, root: TreeNode, sum: int) -> List[List[int]]:
        res, path = [], []
        
        def recur(root, tar):
            if not root: return
            path.append(root.val)
            tar -= root.val
            if tar == 0 and not root.left and not root.right:
                res.append(list(path))
            recur(root.left, tar)
            recur(root.right, tar)
            path.pop()

        recur(root, sum)
        return res

代码来源:mian-shi-ti-34-er-cha-shu-zhong-he-wei-mou-yi-zh-5

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

不吃饭就会放大招

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值