题目描述
输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
示例 1:
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7] Output: [3,9,20,null,null,15,7]
示例 2:
Input: preorder = [-1], inorder = [-1] Output: [-1]
限制:
0 <= 节点个数 <= 5000
本题是要用前序加中序来确定一个二叉树,前序的规则是根左右,所以在前序数组中第一个数肯定是根节点,但是后面哪些是左子树和右子树需要通过中序来确定。中序的规则是左根右,所以根据前序找到根节点,然后在中序数组中找到根节点对应的位置,该位置左侧为左子树、右侧为右子树。二叉树的构建规则大概就是这样,然后就是通过分治的思想来递归地将二叉树建立起来。
递归过程大致如下:
- 首先根据前序数组来找到根节点
- 找到根节点在中序数组中的位置
- 中序数组中根节点左边的都是左子树,右边的都是右子树,然后根据序列长度在前序中找出左子树和右子树的范围。
- 开启左子树和右子树的递归,最后连接到根节点上返回。
过程中有一步是找到根节点在中序数组中的位置,这个可以通过map来实现。代码如下:
class Solution(object):
def buildTree(self, preorder, inorder):
"""
:type preorder: List[int]
:type inorder: List[int]
:rtype: TreeNode
"""
def recur(pos,l_pos,r_pos):
if l_pos > r_pos:
return
node=TreeNode(preorder[pos])
i=m[preorder[pos]]
node.left=recur(pos+1,l_pos,i-1)
node.right=recur(pos+1+i-l_pos,i+1,r_pos)
return node
m,inorder={},inorder
for i in range(len(inorder)):
m[inorder[i]]=i
return recur(0,0,len(preorder)-1)
递归函数的三个参数的意义分别是:当前根节点在先序数组中的位置、当前根节点下所有元素在中序数组中范围的左边界、根节点下所有元素在中序数组中范围的右边界。
进入循环后先判断左右边界是否交叉了,如果交叉说明越过了叶子结点可以直接返回了。
然后建立当前的根节点,找到根节点在中序数组中的位置,也就是 i 。
然后开启左右子树的递归,pos+1是左子树的根节点在前序数组中的位置,pos+1+i-l_pos是右子树的根节点在前序数组中的位置(可以理解为根节点+左子树的序列长度+1)
然后将左右子树连接到根节点上返回。