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

二叉树的前序遍历

递归

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