题目:
剑指 Offer 32 - III. 从上到下打印二叉树 III。
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
链接:https://leetcode-cn.com/leetbook/read/illustrate-lcof/xs0paj/
示例:
例如:
给定二叉树: [3,9,20,null,null,15,7],
思路:
树的广度优先遍历。
首先将上一层节点存入队列,队列不为空时执行循环。遍历本层节点,设置一个临时的列表存储同层节点值,将下一层节点存入队列。
本层节点遍历完后,通过设置布尔变量控制每一层节点值的方向,True表示从左向右,False表示从右向左。将临时列表中的结果追加的返回值列表中,继续下一循环。
推荐与这两道题一起练习:
LC: 剑指 Offer 32 - I. 从上到下打印二叉树
LC: 剑指 Offer 32 - II. 从上到下打印二叉树 II
代码:
# 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]]:
if not root: return []
res, queue, sign = [], [], True
queue.append(root)
while queue:
temp = []
for i in range(len(queue)):
node = queue.pop(0)
temp.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
if sign:
res.append(temp)
sign = False
else:
res.append(temp[::-1])
sign = True
return res
时间复杂度O(n),空间复杂度O(n)。