二叉树的各种遍历方式迭代版(非递归)

class Node:
    '''定义结点'''
    def __init__(self, val):
        self.val = val
        self.left = None
        self.right = None


class Tree:
    def __init__(self):
        self.root = None

    def add(self, item):
        '''层次添加元素'''
        node = Node(item)
        if self.root is None:
            self.root = node
            return
        queue = [self.root]
        while queue:
            cur_node = queue.pop(0)
            if cur_node.left is None:
                cur_node.left = node 
                return
            else:
                queue.append(cur_node.left)
            if cur_node.right is None:
                cur_node.right = node 
                return
            else:
                queue.append(cur_node.right)

    def preorder(self, root):
        '''前序遍历,堆栈,迭代'''
        if not root:
            return
        stack = []
        node = root
        while stack or node:
            while node:
                print(node.val, end=' ')
                stack.append(node)
                node = node.left
            node = stack.pop()
            node = node.right
        print()

    def inorder(self, root):
        '''中序遍历,堆栈,迭代'''
        if not root:
            return
        stack = []
        node = root
        while stack or node:
            while node:
                stack.append(node)
                node = node.left
            node = stack.pop()
            print(node.val, end=' ')
            node = node.right
        print()

    def postorder(self, root):
        '''后续遍历,堆栈,迭代'''
        if root == None:
            return
        stack = []
        res = []
        node = root
        stack.append(node)
        while stack:
            node = stack.pop()
            if node.left:
                stack.append(node.left)
            if node.right:
                stack.append(node.right)
            res.append(node)
        while res: 
            print(res.pop().val, end=' ')
        print()

    def breadth_travel(self, root):
        '''层次遍历'''
        queue = [root]
        while queue:
            cur_node = queue.pop(0)
            print(cur_node.val, end=' ')
            if cur_node.left is not None:
                queue.append(cur_node.left)
            if cur_node.right is not None:
                queue.append(cur_node.right)
        print()


if __name__ == '__main__':
    tree = Tree()
    for i in range(10):
        tree.add(i)
    print('前序遍历:', end='')
    tree.preorder(tree.root)
    print('中序遍历:', end='')
    tree.inorder(tree.root)
    print('后序遍历:', end='')
    tree.postorder(tree.root)
    print('层次遍历:', end='')
    tree.breadth_travel(tree.root)

构建一个如下图所示的二叉树:
在这里插入图片描述
遍历结果如下:
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值