题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
Solution
在常规层序遍历改进而来。使用一个栈保存下一次要访问的各个节点,只是入栈时变换左右节点。
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
if not root: return []
res, queue, = [], collections.deque()
queue.append(root)
while queue:
tmp, queue_next = collections.deque(), collections.deque()
for node in queue:
if (len(res)%2) == 0: tmp.append(node.val)
else: tmp.appendleft(node.val)
if node.left: queue_next.append(node.left)
if node.right: queue_next.append(node.right)
queue = queue_next
res.append(list(tmp))
return res