文章目录
【面试题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