重建二叉树（思路+python代码）

1.题目

输入一棵二叉树前序遍历和中序遍历的结果，请重建该二叉树。

注意:

• 二叉树中每个节点的值都互不相同；
• 输入的前序遍历和中序遍历一定合法；

数据范围

树中节点数量范围 [0,100][0,100]。

样例

给定：
前序遍历是：[3, 9, 20, 15, 7]
中序遍历是：[9, 3, 15, 20, 7]

返回：[3, 9, 20, null, null, 15, 7, null, null, null, null]
返回的二叉树如下所示：
    3
   / \
  9  20
    /  \
   15   7

2.思路

对于本题，我们需要知道前序遍历和中序遍历是什么？

前序遍历：先遍历根节点，然后是左孩子，最后是右孩子

中序遍历：先遍历左孩子，然后是根节点，最后是右孩子

总而言之，前中后是相对于根节点的位置

像

     4
    / \
   3   5
  / \
 7   8

这个二叉树：

前序遍历：4，3，7，8，5

中序遍历：7，3，8，4，5

后序遍历：7，8，3，5，4

了解，前，中，后序遍历后，我们可以知道题中的前序遍历第一个元素便是根节点，而在中序遍历中找到根节点的位置，这个位置前面的全部是根节点的左子树，后面的全是根节点的右子树，我们可以很清晰的了解到左子树有多少个元素，右子树有多少个元素。又因为前序遍历中，是先根再左再右的，同时我们事先知道了左子树元素的个数，因此，我们可以在前序遍历中找到左子树的范围也就是[1:index(root)+1]，这个index(root)是指再中序遍历中的位置，因为切片取不到最后一个元素，且要从1开始（下标为0的是根节点），所以+1，像上面列举的那个二叉树，根节点4再中序遍历中的下标是3，所以在前序遍历中，左子树的元素的下标便是1，2，3，同理，那再前序遍历中右子树的范围便是[index+1:],这样我们分别在中序遍历中和前序遍历中找到了左右子树，接下来便是在左子树中找到对应的子根节点，以及其的左右子树，在右子树中找到对应的子根节点，以及其的左右子树，通过递归，完成这颗树的构建。

3.代码

 

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution(object):
    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: TreeNode
        """
        if not preorder or not inorder:#如果前序遍历和后序遍历其中有一个为空，输出None
            return None
        root=TreeNode(preorder[0])
        t=inorder.index(preorder[0])#找到根节点在中序遍历中的下标
        root.left=self.buildTree(preorder[1:t+1],inorder[:t])#在前序遍历和中序遍历中root的左子树，递归
        root.right=self.buildTree(preorder[t+1:],inorder[t+1:])#在前序遍历和中序遍历中root的右子树，递归
        return root