剑指offer-23 二叉搜索树的后序遍历序列
一、题目描述
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
如果输入如下二叉搜索树:
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
二、解题步骤
1.解题思路
(1)由于路径是从根节点出发到叶子节点的,总是以根节点开始为起点,所以此题需要首先遍历根节点,在树的遍历中,只有前序遍历是首先遍历根节点,所以要首先进行前序遍历。
(2)在前序遍历中,记录根节点到当前节点的路径,路径为根节点到叶子节点的路径,并且各节点的值的和等于目标值时,将此路径加入结果列表。
算法流程:
(1)当前节点为root,目标值为target.
(2) 如果节点root为空,直接返回。
(3)递推工作:
①路径更新:将当前节点的值 node.val 加入路径path中。
② 目标值更新:target=target-node.val
③路径记录:当遍历到的节点node是叶子节点并且此时路径和等于目标值target时,将目前的路径加入到result中。
④ 先序遍历:递归左右子树
⑤ 路径恢复:回溯时的穷举,向上回溯前,需要将当前节点从路径path中删除,执行path.pop()即可。
2.代码实现
# -*- 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):
#result的元素是路径列表,path是路径列表
result,path=[],[]
def recure(node,target):
if not node:
return []
path.append(node.val)
target=target-node.val
if target==0 and not node.left and not node.right:
result.append(list(path))
recure(node.left, target)
recure(node.right, target)
path.pop()
recure(root, expectNumber)
return result