剑指 Offer 07. 重建二叉树

---

一、题目

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

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

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、解法

1.递归

（1）前序的第一个节点是根节点，前序的第二个节点一定是根节点的左节点吗？

不一定？有可能不存在左子树
在前序中确定根节点，然后在中序中找到根节点（无重复元素）。
中序中 根节点的左边都是左子树，右边都是右子树。

（2）右子树的根节点如何确认？

前序： 根—左---右
右子树的根节点下标= （前序中）根节点下标+左子树的长度+1
左子树的长度=(中序中) 根节点下标 - 当前左子树开始下标

（3）划分为子问题

【左子树的根节点，中序中左子树的开始下标，中序中左子树的结束下标】
开始下标：left ； 结束下标：当前根节点的下标-1

【右子树的根节点，中序中右子树的开始下标，中序中右子树的结束下标】
开始下标：当前根节点的下标+1 ； 结束下标：right

（4）递归中止条件
如果当前左边界>右边界：
return NUll

# Definition for a binary tree node.
# 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]) -> TreeNode:
       
        def build(prerootindex,left_start,right_end):
            #递归终止条件
            if left_start>right_end:
                return None

            node=TreeNode(preorder[prerootindex])
            
            #根据 前序 找到      中序根节点下标
            inrootindex=index_map[preorder[prerootindex]]
             
            #左子树的 前序根节点 ，中序的区间开始，中序的区间结束
            node.left=build(prerootindex+1,left_start,inrootindex-1) 
            #右子树的 前序根节点，  前序根节点下标+左子树长度(inrootindex-left_start)+1
            #中序的区间开始，中序的区间结束
            node.right= build(prerootindex+inrootindex-left_start+1,inrootindex+1,right_end)
            
            return node

         #对中序进行遍历，构建哈希映射，使值和下标对应
        index_map={element:i for i,element in enumerate(inorder)}
        #前序的根节点,中序区间开始，中序区间结束
        tree=build( 0,0,len(inorder)-1)
        return tree

每次找出根节点都需要遍历一次中序，可以使用map将值与下标进行匹配

 index_map={element:i for i,element in enumerate(inorder)}