leetcode--重建二叉树（递归）

题目来源：链接

题目描述：

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

 

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

实现代码：

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        self.dic, self.po = {}, preorder
        for i in range(len(inorder)):
            self.dic[inorder[i]] = i
        return self.recur(0,0,len(inorder)-1)
    
    def recur(self,pre_root,in_left,in_right):
        if in_left > in_right:
            return None
        root = TreeNode(self.po[pre_root])
        i = self.dic[self.po[pre_root]]
        root.left = self.recur(pre_root+1,in_left,i-1)
        root.right = self.recur(i-in_left+pre_root,i+1,in_right)
        return root

解题思路：

先根据先序遍历找到当前树（或者子树）的根节点，字典存放中序遍历每个元素的下标

在用这个找到的根节点在中序遍历找到位置，然后分左右子树

在递归到左子树，递归到右子树