输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
这题是深度优先搜索,它要求经过结点数值之和为一个固定值,所以经过一个结点就要从要求的和里把当前结点的值减下去。
# -*- 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
if not root:
return []
# not root.left and not root.right保证结点是叶子结点
if not root.left and not root.right and expectNumber == root.val:
return [[root.val]]
res = []
# 分别在左右子树找路径
# 按照根左右的深度搜索
left = self.FindPath(root.left, expectNumber-root.val)
right = self.FindPath(root.right, expectNumber-root.val)
# 把所有的路径存在res里
# 递归返回的路径
for i in left+right: # [[A]]+[[B]] = [[A],[B]]
# 把当前的根结点放进去
res.append([root.val]+i)
return res
递归调用的特别之处是调用的函数自己本身,但是调用的过程和正常的调用是一样的,进入递归调用之后就相当于进入一个新的函数,和之前语句所处的位置没有关系。