一、题目描述
给你二叉树的根节点 root
,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
示例 1
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[20,9],[15,7]]
示例 2
输入:root = [1]
输出:[[1]]
示例 3
输入:root = []
输出:[]
提示:
树中节点数目在范围 [0, 2000] 内
-100 <= Node.val <= 100
二、代码
代码如下:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def zigzagLevelOrder(self, root: Optional[TreeNode]) -> List[List[int]]:
if root is None:
return []
result = []
# 最开始的一层中只有根节点一个节点
cur = [root]
# 从左往右:0 ;从右往左:1
direction = 0
while cur:
next = []
path = []
for node in cur:
# path 用于存储这一层所有的结点值
path.append(node.val)
# 如果当前节点有左子树
if node.left:
# 则在下一层需要遍历的结点数组中添加
next.append(node.left)
# 如果当前节点有右子树
if node.right:
next.append(node.right)
# 让当前遍历的结点数组更新为下一次遍历的结点数组,即下一层所有结点的数组
cur = next
if direction == 0:
result.append(path)
direction = 1
else:
path.reverse()
result.append(path)
direction = 0
return result
三、解题思路
本题是对 102. 二叉树的层序遍历 Python的一个加深,其基本原理还是通过二叉树的层序遍历,唯一的不同是需要区分当前层是从左往右的顺序还是从右往左的顺序,对应存储数组path
则是正序和倒序。
所以本题解只需要对每层的元素判断其输入方向即可,这里用direction
来表示方向,direction=0
表示从左往右读取(即path
数组正序);direction=1
表示从右往左读取(即path
数组倒序)。
开始的时候是从左往右的顺序进行遍历,然后下一层变成从右往左,然后再下一层又变成从左往右…每次需要从右往左遍历时,只需要将当前遍历数组逆序即可,这里采用reverse()
函数。