二叉树问题---先序，中序，和后序数组两两结合重构二叉树

最新推荐文章于 2025-05-25 10:30:23 发布

最新推荐文章于 2025-05-25 10:30:23 发布 · 563 阅读

文章标签：

#二叉树 #重构

数据结构与算法专栏收录该内容

168 篇文章

订阅专栏

本文介绍如何使用不同类型的数组（如先序、中序、后序）来重构二叉树，并提供具体的Python实现代码。文章重点讲解了在特定条件下如何通过先序、中序和后序遍历来重建一棵完整的二叉树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

注意：
　　如果一棵二叉树除叶节点之外，其他的节点都有左孩子和右孩子，这样的树才能被先序和后序数组重建

#python3.5
#先序，中序数组重构二叉树
def preInToTree(pre, mid):
    def preIn(pre, pi, pj, mid, mi, mj, map):
        if pi > pj:
            return None
        head = TreeNode(pre[pi])
        index = map.get(pre[pi])
        head.left = preIn(pre, pi+1, pi+index-mi, mid, mi, index-1, map)
        head.right = preIn(pre, pi+index-mi+1, pj, mid, index+1, mj, map)
        return head

    if pre == None or mid == None or len(pre) != len(mid):
        return None
    map = {}
    for i in range(len(mid)):
        map[mid[i]] = i
    return preIn(pre, 0, len(pre)-1, mid, 0, len(mid)-1, map)

#中序，后序数组重构二叉树
def inPosToTree(mid, pos):
    def inPos(mid, mi, mj, pos, si, sj, map):
        if si > sj:
            return None
        head = TreeNode(pos[sj])
        index = map.get(pos[sj])
        head.left = inPos(mid, mi, index-1, pos, si, si+index-mi-1, map)
        head.right = inPos(mid, index+1, mj, pos, si+index-mi, sj-1, map)
        return head


    if mid == None or pos == None or len(mid) != len(pos):
        return None
    map = {}
    for i in range(len(mid)):
        map[mid[i]] = i
    return inPos(mid, 0, len(mid)-1, pos, 0, len(pos)-1, map)

#先序，后序数组重构二叉树
#一棵二叉树除叶子节点外，其他所有的节点都有左孩子和右孩子，才能被先序和后序数组重构出来
def prePosToTree(pre, pos):
    def prePos(pre, pi, pj, pos, si, sj, map):
        head = TreeNode(pre[pi])
        if pi == pj:
            return head
        pi += 1
        index = map.get(pre[pi])
        head.left = prePos(pre, pi, pi+index-si, pos, si, index, map)
        head.right = prePos(pre, pi+index-si+1, pj, pos, index+1, sj-1, map)
        return head

    if pre == None or pos == None or len(pre) != len(pos):
        return None
    map = {}
    for i in range(len(pos)):
        map[pos[i]] = i
    return prePos(pre, 0, len(pre)-1, pos, 0, len(pos)-1, map)