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

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

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

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

1、题目分析

找出一个树种所有的路径和等于sum的集合。既然是二叉树肯定需要使用递归的方法去解决。

2、解题分析

• 初始化两个空数组，一个是结果集数组；另一个是路径数组
• 每次将路径添加到路径数组中的时候就减少sum；
  • 如果sum==0的时候且这个节点是根节点就把它添加到结果集数组中
• 分别这样递归的去遍历左子树和右子树
• 每一层的递归结束之后都要把路径往出弹出一个
• 如果root为空直接return返回

3、代码

# 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 []

        res = []
        temp = []
        def helper(root,tar):
            if not root:
                return
            temp.append(root.val)
            tar-=root.val
            if tar==0 and not root.left and not root.right:
                res.append(list(temp))
            helper(root.left,tar)
            helper(root.right,tar)
            temp.pop()
        
        helper(root,sum)
        return res

总结 递归的难点在于涉及递归的入口及结束条件。