从先序与中序遍历序列构造二叉树

从先序与中序遍历序列构造二叉树

描述：
给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

递归法

解题思路：
通过先序遍历我们可以找到root，因为列表中没有重复数字的原因，根据root我们可以根据中序找到root的索引位置，该索引位置左边就是左子树，右边就是右子树。而且根据中序中root的索引位置，可以在先序中确定先序里左子树索引位置为[1:1+idx)，右子树索引位置[1+idx:-1]。根据位置，通过递归法直到遍历到结束。

ps:图片来源leetcode题解

代码实现：

class TreeNode:
	def __init__(self,x):
		self.val = x
		self.left = None
		self.right = None
class Solution:
	def buildTree(self,preorder:List[int],inorder:List[int])->:
		# 递归中止条件
		if not preorder or not inorder:
			return
		
		# 根据先序列表得到根节点root
		root = TreeNode(preorder[0])
		# 根据中序列表得到根节点的索引位置
		idx = inorder.index(preorder[0])
		
		# 递归法找到根节点的左子树部分
		root.left = self.buildTree(preorder[1:1+idx],inorder[0:idx])
		# 递归法找到根节点的右子树部分
		root.right = self.buildTree(preorder[1+idx:],inorder[idx+1:])
		return root

另一道题目，从中序与后序遍历序列构造二叉树，可以点击链接查看另一篇文章