python 二叉树遍历

最新推荐文章于 2024-10-21 08:57:50 发布

转载最新推荐文章于 2024-10-21 08:57:50 发布 · 221 阅读

文章标签：

#二叉树遍历

数据结构专栏收录该内容

5 篇文章

订阅专栏

二叉树的前序遍历

递归

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

转载自：https://blog.youkuaiyun.com/wenkenza5368/article/details/79573333