《剑指offer》刷题系列——（三十五）二叉树中和为某一值的路径

题目

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

思路

回溯法，先序遍历
既要记录每条路径上的节点，也要更新当前sum值
rec(root,sum)函数：
先访问根节点root，然后把根节点加入到path[ ]中保存，同时更新sum=sum-root.val，然后从root的左子树和右子树中寻找符合当前sum值得路径。
临界条件：
当root为空时，说明访问到了叶子结点的的孩子节点，返回；
当sum的值更新为0，并且当前访问到了叶子结点，说明找到了一条符合条件的路径，将path保存在res列表中。
注意：每次返回时都要从path中把当前节点删除。

代码

# 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]]:
        if not root: return []
        path = []
        res = []
        return self.rec(root,sum,res,path)
    def rec(self,root,sum,res,path):
        if not root: return 
        path.append(root.val)
        sum-=root.val
        if sum==0 and not root.left and not root.right:
            res.append(list(path))
        self.rec(root.left,sum,res,path)
        self.rec(root.right,sum,res,path)
        path.pop()
        return res

复杂度

时间复杂度 O(N) ： N为二叉树的节点数，先序遍历需要遍历所有节点。
空间复杂度 O(N) ： 最差情况下，即树退化为链表时，path 存储所有树节点，使用 O(N) 额外空间。

知识点

值得注意的是，记录路径时若直接执行 res.append(path) ，则是将 path 对象加入了 res ；后续 path 改变时， res 中的 path 对象也会随之改变。
正确做法：res.append(list(path)) ，相当于复制了一个 path 并加入到 res 。