剑指Offer刷题笔记——重建二叉树

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

前序是根左右，中序是左根右。从前序依次拿出的结点都是它所在子树的根节点，在中序这个结点的左右就是它的左右子树。递归执行知道子树都变成结点。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回构造的TreeNode根节点
    def reConstructBinaryTree(self, pre, tin):
        # write code here
        if len(pre) == 0:
            return None
        elif len(pre) == 1:
            return TreeNode(pre[0])
        else:
            # 从前序头部拿出来的肯定是根节点
            root = TreeNode(pre[0])
            # pos 是中序里根节点的位置
            pos = tin.index(pre[0])
            # tin[:pos]这些是左子树的所有结点
            # pre[1:pos+1] 同样一个子树它的结点数目是一致的，数目是pos个，但是前序第一个结点是根节点。所以前序中的左子树是从1到pos+1
            root.left = self.reConstructBinaryTree(pre[1:pos+1], tin[:pos])
            # 右子树同理
            root.right = self.reConstructBinaryTree(pre[pos+1:], tin[pos+1:])
        return root