二叉树的四种遍历(非递归)

class treeNode:
    def __init__(self, val, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
        
def first_order(root):
    if not root:
        return
    node = root
    stack = []
    while stack or node:
        while node:
            #一直向左遍历,将遍历到的结点都压入这个栈
            print(node.val)
            stack.append(node)
            node = node.left
        node = stack.pop()
        node = node.right

def mid_order(root):
    if not root:
        return
    node = root
    stack = []
    while node or stack:
        while node:
            stack.append(node)
            node = node.left
        node = stack.pop()
        print(node.val)
        node = node.right
    
def last_order(root):
    if not root:
        return
    node = root
    stack1 = []
    stack2 = []
    stack1.append(node)
    while stack1: 
        # 该循环用来找出后序遍历的逆序,存放在stack2里
        node = stack1.pop()
        if node.left:
            stack1.append(node.left)
        if node.right:
            stack1.append(node.right)
        stack2.append(node)
    while stack2:
        print(stack2.pop().val)

def layer_order(root):
    # 宽度优先,用队列实现
    if not root:
        return
    ls = [root]
    while ls:
        node = ls.pop(0)
        print(node.val)
        if node.left:
            ls.append(node.left)
        if node.right:
            ls.append(node.right)
            

# 测试用例
b = treeNode(2, left=treeNode(4), right=treeNode(5))
c = treeNode(3)
root = treeNode(1, b, c)
first_order(root)
mid_order(root)
last_order(root)
layer_order(root)

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值