题目:
给定两个整数数组
preorder
和inorder
,其中preorder
是二叉树的先序遍历,inorder
是同一棵树的中序遍历,请构造二叉树并返回其根节点。
示例 1:
输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7] 输出: [3,9,20,null,null,15,7]示例 2:
输入: preorder = [-1], inorder = [-1] 输出: [-1]提示:
1 <= preorder.length <= 3000
inorder.length == preorder.length
-3000 <= preorder[i], inorder[i] <= 3000
preorder
和inorder
均 无重复 元素inorder
均出现在preorder
preorder
保证 为二叉树的前序遍历序列inorder
保证 为二叉树的中序遍历序列
解题思路:
首先如果先序遍历序列或者中序遍历序列为空,则直接返回空。然后根据先序遍历首先遍历根节点来创建根节点,再根据根节点获得在中序遍历中对应根节点的下标。最后根据递归来得到二叉树的左右子树并返回得到的二叉树。
程序代码:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def buildTree(self, preorder, inorder):
"""
:type preorder: List[int]
:type inorder: List[int]
:rtype: TreeNode
"""
if not preorder or not inorder:
return None
root=TreeNode(preorder[0])
index=inorder.index(preorder[0])
root.left=self.buildTree(preorder[1:1+index],inorder[:index])
root.right=self.buildTree(preorder[1+index:],inorder[index+1:])
return root
运行结果: