剑指offer-24.二叉树中和为某一值的路径-Python

24.二叉树中和为某一值的路径

题目描述

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

记录

深度优先遍历：DFS

1. 到达叶子节点且该叶子节点满足最后一个数，返回val
2. 非叶子节点，继续遍历，分别对根节点的左子树和右子树进行递归，直到找到叶子结点。并加上根节点，形成最后的集合；
3. 有则返回路径，没有则返回[]。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回二维列表，内部每个列表表示找到的路径
    def FindPath(self, root, expectNumber):
        # write code here
        result = []
        #常规判空
        if root is None:
            return []
        #到达叶子节点且该叶子节点满足最后一个数
        if not root.left and not root.right and root.val == expectNumber:
            return [[root.val]]
        #非叶子节点，继续遍历
        else:
        	#左右子树遍历结果的集合
            left = self.FindPath(root.left, expectNumber - root.val)
            right = self.FindPath(root.right, expectNumber - root.val)
            #加上根节点，形成最后的集合
            for i in left + right:
                result.append([root.val]+i)
            return result

相似，只是定义一个子函数：

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回二维列表，内部每个列表表示找到的路径
    def FindPath(self, root, expectNumber):
        self.res = []
        #判空，返回[]
        if not root:
            return []
        self.subPath(root,[],expectNumber)
        return self.res
        
    def subPath(self, root, path,number):
    	#到达叶子节点且该叶子节点满足最后一个数
        if not root.left and not root.right:
            if number == root.val:
                self.res.append(path+[root.val])
        #左右子树遍历结果的集合，加上根节点，形成最后的集合        
        if root.left:
            self.subPath(root.left,path+[root.val],number-root.val)
        if root.right:
            self.subPath(root.right,path+[root.val],number-root.val)