题目
之字形打印二叉树
一、思路
基于上一题的基础上,我们只是每次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