根据前序和中序遍历构造一颗二叉树(python)

假设现在有这样两个遍历结果：
中序：D B E A F C
后序：A B D E C F

在前序遍历结果中，最左边的元素就是二叉树的根节点。所以在上面的例子里面，‘A’ 就是想构造的树的根节点。在中序遍历的结果里，找到’A’ 元素。‘A’ 左边的所有元素，属于以’A’ 为根的树的左子树，‘A’ 的右边的所有元素，属于以’A’ 为根的树的右子树。如下图所示：

然后我们再递归的遵循上面的步骤，得到下图中的树。

算法伪代码：

1.在前序遍历中，挑选出preIndex 位置的元素。然后preIndex 自增1。preIndex 初始化为0；
2. 用挑选到的这个元素，创建一个新的节点；
3. 找到这个元素在中序遍历中的位置。用inIndex表示下标；
4. 利用inIndex左边的元素，递归调用。构造节点的左子树；
5. 利用inIndex右边的元素，递归调用。构造节点的右子树；
6. 返回该节点。
实现代码：

class Node:
    def __init__(self, data):
        self.data = data
        self.left = None
        self.right = None

def buildTree(inOrder, preOrder, inStrt, inEnd):
    if (inStrt > inEnd):
        return None

    # 从Preorder 中挑选preIndex位置元素，preIndex自增 1
    tNode = Node(preOrder[buildTree.preIndex])
    buildTree.preIndex += 1

    # 如果节点没有孩子则返回
    if inStrt == inEnd:
        return tNode

    # 否则，找到这个元素在中序遍历中的位置
    inIndex = inOrder.index(tNode.data)
    # 利用找到的这个位置，递归地创建左右子树
    tNode.left = buildTree(inOrder, preOrder, inStrt, inIndex - 1)
    tNode.right = buildTree(inOrder, preOrder, inIndex + 1, inEnd)

    return tNode


def printPostorder(node):
    if node is None:
        return

    printPostorder(node.left)
    printPostorder(node.right)
    print(node.data)

inOrder = ['D', 'B', 'E', 'A', 'F', 'C']
preOrder = ['A', 'B', 'D', 'E', 'C', 'F']
# 静态变量 preIndex，初始化为 0
buildTree.preIndex = 0
root = buildTree(inOrder, preOrder, 0, len(inOrder) - 1)
# 打印后序遍历结果
printPostorder(root)