JZ59 按之字形顺序打印二叉树

描述

给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
例如:
给定的二叉树是{1,2,3,#,#,4,5}

 
该二叉树之字形层序遍历的结果是[[1],[3,2],[4,5]]。

好像是层次遍历,但是因为层次遍历每一层都是从左向右,这里要进行分层处理,分奇偶层。

官方题解:层次遍历+栈:https://blog.nowcoder.net/n/cf46d04c5c98413086ab604268ebaeda?f=comment

设置一个栈res、一个辅助栈stack,还有一个临时列表tmp,tmp用来临时存储当前层打印结果。

   根据index判断所在层是奇数层还是偶数层:

        1、若为奇数层:将tmp直接添加到res中

        2、若为偶数层:将tmp倒序添加到res中

class Solution:
    def Print(self, pRoot):
        # write code here
        if not pRoot:
            return []
        res = []
        # 辅助栈
        stack = [pRoot]
        # 记录层数
        index = 1
        while stack:
            tmp = []
            for i in range(len(stack)): 
                node = stack[0]
                stack = stack[1:]
                tmp.append(node.val)
                if node.left:
                    stack.append(node.left)
                if node.right:
                    stack.append(node.right)
            # 判断所在树的层数,根据层数对 tmp 是否倒序存储
            if index % 2 == 0:
                res.append(tmp[::-1])
            else:
                res.append(tmp)
            index += 1 
        return res

 还在评论区看到可以用双端队列,但是只看到JAVA版本。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值