问题描述:
返回与给定的前序和后序遍历匹配的任何二叉树。
pre
和
post
遍历中的值是不同的正整数。
示例:
输入:pre = [1,2,4,5,3,6,7], post = [4,5,2,6,7,3,1]
输出:[1,2,3,4,5,6,7]
提示:
1 <= pre.length == post.length <= 30
pre[]
和post[]
都是1, 2, ..., pre.length
的排列- 每个输入保证
至少
有一个答案。如果有多个答案,可以返回其中一个。
问题分析:
我们知道,中序和先序、后续,可以唯一确定一个二叉树,而先序和后续,就不一定了。因为返回一个即可,所以不用考虑太繁杂。从整个序列来看看,pre
第一个元素和post
最后一个元素是根节点,所以,可以从pre
开始,进行遍历,依次作为根节点,与此同时,确定一个这个根节点在post
中的位置,获取两个区间,左右深度递归。
Python3实现:
class Solution:
def constructFromPrePost(self, pre, post):
if not pre: return None
root = TreeNode(pre[0]) # 创建当前根节点
if len(pre) == 1: return root # 如果只有一个值,直接返回
i = post.index(pre[1]) + 1 # 获取当前跟节点,在 post 中的位置
root.left = self.constructFromPrePost(pre[1:i+1], post[:i]) # 递归左子树
root.right = self.constructFromPrePost(pre[i+1:], post[i:]) # 递归右子树
return root
声明: 总结学习,有问题可以批评指正,大神可以略过哦。