返回与给定的前序和后序遍历匹配的任何二叉树。
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 的排列
每个输入保证至少有一个答案。如果有多个答案,可以返回其中一个。
tips:事实上,根据前序和后序构建的二叉树不唯一,理由是前序与后序都没有明确规定节点间的父子关系,例如下图所示:
还是可以根据一般的思路,采用递归思想,对于每一个先序序列,划分出对应的根节点、左子树、右子树范围即可自上而下构建出二叉树。
例如对于上例中的先序序列[1,2,4,5,3,6,7],第一个节点一定为根节点,第2到第i个节点为左子树,第i+1到最后一个节点为右子树,那么问题就可以简化为:如何确定左右子树分界点?
即 前序遍历的第二个元素代表左子树的根节点,位于后序遍历左子树的最后一个节点。那么左子树的长度在后序遍历中为:post.index(pre[1])+1 。代码如下:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def constructFromPrePost(self, pre: List[int], post: List[int]) -> TreeNode:
if not pre or not post:
return
root=TreeNode(pre.pop(0))
if pre:
index=post.index(pre[0])+1
root.left=self.constructFromPrePost(pre[:index],post[:index])
root.right=self.constructFromPrePost(pre[index:],post[index:-1])
return root