Code练习笔记5：重建二叉树

1. 题目内容

输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

示例 1:

---------------------------------------------------------------------------------------------------------------------------------

来源：力扣（LeetCode）
 

2. 解题思路

对于任意一颗树而言，

前序遍历的形式总是 根节点>>左子树的前序遍历结果>>右子树的前序遍历结果,即根节点总是前序遍历中的第一个节点。

中序遍历的形式总是 左子树的中序遍历结果>>根节点>>右子树的中序遍历结果。

只要我们在中序遍历中定位到根节点，那么我们就可以分别知道左子树和右子树中的节点数目。由于同一颗子树的前序遍历和中序遍历的长度显然是相同的，因此我们就可以对应到前序遍历的结果中，对上述形式中的所有左右括号进行定位。这样的话，我们就知道了左子树的前序遍历和中序遍历结果，以及右子树的前序遍历和中序遍历结果，我们就可以递归地对构造出左子树和右子树，再将这两颗子树接到根节点的左右位置。

3. python实现

class Solution:
	def buildTree(self, preorder: List[int], inorder: List[int]):
		def myBuildTree(preorder_left: int, preorder_right: int, inorder_left: int, inorder_right： int):
			if preorder_left > preorder_right:
				return None

			# 前序遍历中的第一个节点就是根节点
			preorder_root = preorder_left

			#在中序遍历中定位根节点
			inorder_root = index[preorder[preorder_root]]

			#先把根节点建立出来
			root = TreeNode(preorder[preorder_root])
			#得到左子树中的节点数目
			size_left_subtree = inorder_root - inorder_left
			#递归构造左子树，并连接到根节点
			root.left = myBuildTree(preorder_left + 1, preorder_left + size_left_subtree, inorder_left, inorder_root - 1)
			#递归构造右子树，并连接到根节点
			root.right = myBuildTree(preorder_left + size_left_subtree + 1, preorder_right, inorder_root + 1, inorder_right)
			return root 

		n = len(preorder)
		#构造哈希映射，快速定位根节点
		index = {element: i for i, element in enumerate(inorder)}
		return myBuildTree(0, n-1, 0, n-1)

4. 复杂度

时间复杂度：O(n)，其中 n 是树中的节点个数。

空间复杂度：O(n)。