剑指offer实践 ——32.3.之字形打印二叉树(python版)

文章目录



题目

之字形打印二叉树
在这里插入图片描述


一、思路

基于上一题的基础上,我们只是每次print的方向改变了。而collections的deque,这个队列既能队头插入、也能队尾插入。既能队头取出,也能队尾取出。
如果设计一个flag,每当换行时,我们就切换取值和放入的方式,那就好了~
如果时正常顺序打印的层,可以和上一题保持一样的操作。
但是如果是相反的顺序打印这个层:
在这里插入图片描述
上一层保存的节点应该是:[4,5,6,7]
此时我们首先需要打印的是最后一个节点的右节点15,应该从上一层队尾取出节点7,依次print节点7的右孩子、左孩子。
再从上层的队尾取出节点6,依次print节点6的右孩子、左孩子。

def print_top_to_bottom_tree(tree):
    if not tree:
        return []
    res,queue = [], collections.deque()
    queue.append(tree)

    flag = 0
    while queue:
        tmp = []
        for i in range(len(queue)):
            if flag % 2 == 0:
                node = queue.pop()
                if node.right:
                    queue.appendleft(node.right)
                if node.left:
                    queue.appendleft(node.left)
            else:
                node = queue.popleft()
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
            tmp.append(node.root)
        flag += 1
        res.append(tmp)
    return res
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值