python实现二叉树的操作（7种）

最新推荐文章于 2025-04-06 16:15:21 发布

wenkenza5368

最新推荐文章于 2025-04-06 16:15:21 发布

阅读量2.9k

点赞数

本文链接：https://blog.youkuaiyun.com/wenkenza5368/article/details/79573333

版权

本文详细介绍了二叉树的前序、中序和后序遍历算法，包括递归和迭代两种实现方式，并提供了层次遍历的迭代实现。每种遍历方式都附有清晰的代码示例。

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

二叉树的前序遍历

递归

def preorder(root,res=[]):
    if not root:
        return 
    res.append(root.val)
    preorder(root.left,res)
    preorder(root.right,res)
    return res

迭代

def preorder(root):
    res=[]
    if not root:  
        return []
    stack=[root]
    while stack:
        node=stack.pop()
        res.append(node.val)
        if node.right:
            stack.append(node.right)
        if node.left:
            stack.append(node,left)
    return res

二叉树的中序遍历

递归

def inorder(root,res=[]):
    if not root:
        return 
    inorder(root.left,res)
    res.append(root.val)
    inorder(root.right,res)
    return res

迭代

def inorder(root):
    stack=[]
    node=root
    res=[]
    while stack or node:
        while node:
            stack.append(node)
            node=node.left
        node=stack.pop()
        res.append(node.val)
        node=node.right
    return res

二叉树的后序遍历

递归

def laorder(root,res=[]):
    if not root:
        return 
    laorder(root.left,res)
    laorder(root.right,res)
    res.append(root.val)
    return res

迭代

def laorder(root):
    stack=[root]
    res=[]
    while stack:
        node=stack.pop()
        if node.left:
            stack.append(node.left)
        if node.right:
            stack.append(node.right)
        res.append(node.val)
    return res[::-1]

二叉树的层次遍历

迭代

def levelorder(root):
    queue=[root]
    res=[]
    while queue:
        node=queue.pop(0)
        if node.left:  
            queue.append(node.left)
        if node.right:
            queue.append(node.right)
        res.append(node.val)
    return res