Leetcode做题日记:103. 二叉树的锯齿形层次遍历(PYTHON)

给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

例如:

给定二叉树 [3,9,20,null,null,15,7],

3

/
9 20
/
15 7

返回锯齿形层次遍历如下:

[
[3],
[20,9],
[15,7]
]

第一次的代码:
在上一题的基础上添加一些代码就成,既然是锯齿形,则和我们选择左树还是右树作为开始有关。通过count记录这一层该从左树开始还是从右树开始遍历,而每一次我们都要当前层节点取反,即tr=tr[::-1],因为我们正常来说加入的顺序和锯齿下一层开始的节点顺序是相反的。

	count=0
        if not root:
            return []
        tr=[root]
        tv=[]
        av=[]
        while tr:
            ind=count%2

            nr=[]
            for i in tr:
                if ind:#每层right,left开始不一样
                    if i.right:
                        nr.append(i.right)
                    if i.left:
                        nr.append(i.left)
                else:
                    if i.left:
                        nr.append(i.left)
                    if i.right:
                        nr.append(i.right)
                tv.append(i.val)
            tr=nr[:-1]#每次取反
            av.append(tv)
            tv=[]
            count+=1
        return av

28ms,排名90%
当count=0时,看起来是先取左再取右,其实在tr=nr[::-1]这一步就把结果反过来了所以正确。

当然也可以写的更紧凑一些:用k=-k来记录因该反转的层,其实细心的朋友可以发现,上一题按层添加的时候,每层都是按顺序,只要相应的间隔取反就行了。

	k=1
        if not root:
            return []
        tr=[root]
        tv=[]
        av=[]
        while tr:
            nr=[]
            for i in tr:
                if i.right:
                    nr.append(i.right)
                if i.left:
                    nr.append(i.left)
                tv.append(i.val)
            k=-k
            tr=nr[:]
            av.append(tv[::k])#间隔一个取反一次
            tv=[]
        return av

24ms,排名96%

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值