python 已知前序遍历和中序遍历求二叉树结构代码，用python实现，二叉树算法

医学小达人

于 2023-07-10 15:00:30 发布

阅读量322

点赞数

分类专栏：数据结构 Python数据基础文章标签： python 算法

本文链接：https://blog.youkuaiyun.com/L_goodboy/article/details/131639192

版权

Python数据基础同时被 2 个专栏收录

59 篇文章

订阅专栏

数据结构

22 篇文章

订阅专栏

该代码示例展示了如何使用Python通过已知的前序遍历和中序遍历序列来构建二叉树。首先定义了一个TreeNode类，然后实现了一个buildTree函数，它递归地根据给定的遍历序列创建树结构。最后，提供了一个inorderTraversal函数用于验证生成的二叉树的中序遍历是否正确。

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

二叉树算法：已知前序遍历和中序遍历求二叉树代码，用python实现：

# -------------------------------------------------------------------------------
# coding:utf-8
# Description:  
# Reference:
# Author: dacongming
# Date:   2023/7/10
# -------------------------------------------------------------------------------

class TreeNode:
    def __init__(self,val = 0,left = None, right = None):
        self.val = val
        self.left = left
        self.right = right


def buildTree(preorder, inorder):
    if not preorder or not inorder:
        return None
    root_val = preorder[0]
    root = TreeNode(root_val)

    root_index = inorder.index(root_val)

    left_inorder = inorder[:root_index]
    right_inorder = inorder[root_index+1:]

    left_preorder = preorder[1:1+len(left_inorder)]
    right_preorder = preorder[1+len(left_inorder):]

    root.left = buildTree(left_preorder, left_inorder)
    root.right = buildTree(right_preorder, right_inorder)
    return root


preorder = [3, 9, 20, 15, 7]
inorder = [9, 3, 15, 20, 7]
root = buildTree(preorder, inorder)

测试代码：

# 验证结果
def inorderTraversal(root):
    if not root:
        return []

    result = []
    stack = []
    curr = root

    while curr or stack:
        while curr:
            stack.append(curr)
            curr = curr.left
        curr = stack.pop()
        result.append(curr.val)
        curr = curr.right

    return result

print(inorderTraversal(root))  # 输出 [9, 3, 15, 20, 7]