描述
给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
例如:
给定的二叉树是{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版本。