请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
提示:
节点总数 <= 1000
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
res,node_queue,index_queue = [],collections.deque(),collections.deque()
# 根节点索引为0
index_queue.append(0)
node_queue.append(root)
self.bfs(index_queue, node_queue, res)
return res
def bfs(self, index_queue, node_queue, res):
if index_queue or node_queue:
index = index_queue.popleft()
node = node_queue.popleft()
if node is not None:
# 如果索引小于res的长度,说明res已经存储对应层数的数据
# 层数是偶数的时候是从左到右打印,则往偶数层增加成员时候,都是在尾部插入元素
# 奇数的时候从右往左打印,则往奇数层增加成员时候,都是在头部插数据
if index < len(res):
item = res[index]
if index % 2 == 0:
item.append(node.val)
else:
item.insert(0, node.val)
else:
item = []
item.append(node.val)
res.append(list(item))
# 层数的索引加1
index += 1
# 二叉树,一次会添加两个成员
index_queue.append(index)
node_queue.append(node.left)
index_queue.append(index)
node_queue.append(node.right)
self.bfs(index_queue, node_queue, res)
else:
return []