剑指 Offer 07. 重建二叉树

对着前序遍历的数组看，对应到中序遍历的的数组中查找分割点

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        idxs = {}  #节点:该节点在inorder中的索引
        for idx, num in enumerate(inorder):
            idxs[num] = idx
        def recur(root_idx, left, right): #第一个参数，是子树的头节点，是在preorder中的索引。第二三个是该树节点的范围，在inorder中,是连续的。
            if left > right:
                return
            node = TreeNode(preorder[root_idx])
            seg_idx = idxs[preorder[root_idx]]
            #左子树的头节点就在root + 1
            node.left = recur(root_idx + 1, left, seg_idx - 1)
            #找右子树的头节点，位置是root + 左子树的大小 + 1
            node.right = recur(root_idx + seg_idx - left + 1, seg_idx + 1, right)
            return node
        ans = recur(0, 0, len(preorder) - 1)
        return ans