《剑指Offer（第二版）》面试题34. 二叉树中和为某一值的路径

今晚打老虎

于 2020-03-30 16:43:21 发布

阅读量126

点赞数

分类专栏： LeetCode点滴

本文链接：https://blog.csdn.net/qq_24502469/article/details/105202664

版权

LeetCode点滴专栏收录该内容

195 篇文章 2 订阅

订阅专栏

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

示例:
给定如下二叉树，以及目标和 sum = 22，
              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:
[
   [5,4,11,2],
   [5,8,4,5]
] 

提示：
节点总数 <= 10000

按照先序遍历，当遍历到叶节点且路径和达标，保存
如果某个节点的左右孩子遍历完了，从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, sums: int) -> List[List[int]]:
        paths, path = [], []

        def helper(root):
            if not root:
                return 
            path.append(root.val)

            if not root.left and not root.right and sum(path) == sums:
                paths.append(path[:])
            helper(root.left)
            helper(root.right)
            path.pop()

        helper(root)
        return paths