一、题目
给你二叉树的根节点 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]]:
res = []
if not root:
return res
q = deque()
q.append(root)
while q:
cnt = len(q)
pp = []
for _ in range(cnt):
tmp = q.popleft()
pp.append(tmp.val)
if tmp.left: q.append(tmp.left)
if tmp.right: q.append(tmp.right)
res.append(pp.copy())
for i in range(len(res)):
if i % 2 != 0:
res[i] = res[i][::-1]
return res