1.题目
这道题是2024-2-16的签到题,题目难度为中等。
考察知识点为BFS算法和双端队列。
题目链接:二叉树的锯齿形层序遍历
给你二叉树的根节点 root
,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
2.思路
这道题其实归根结底还是二叉树的层序遍历,只不过在遍历的过程中有一个锯齿形的遍历顺序,其实这题我们可以在BFS遍历的基础上模拟遍历,我们肯定玩过冰火人大冒险这个游戏,其中第一关就是类似这种锯齿形的遍历。
正常的层序遍历是每次遍历当前的结点队列,并将它们的子结点添加到下一次的结点队列中,因此这里我们仍然可以借助这种思想,只不过我们加一个判断,判断它是奇数层还是偶数层(假设根节点是奇数层),我的思路如下:
如果当前层是奇数层,我依旧是正常遍历(从左往右),然后我需要先判断它的左子结点,然后再判断右子结点,如果有不为空的,我需要借助deque(双端队列)将它插入到队列左端。这里是为了保证偶数层最右边的结点在队列左侧,最左边的结点在队列右侧。
如果当前层是偶数层,我还是正常遍历,因此前面在奇数层的时候对这一层的节点队列进行了处理,保证我的结点时从右边开始,只不过这次我的判断顺序发生变化,我需要先判断它的右子结点,然后再判断左子结点,如果有不为空的,我需要借助deque(双端队列)将它插入到队列左端。这里是为了保证奇数层最左边的结点在队列左侧,最右边的结点在队列右侧,即顺序归位。
3.代码
# 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]]:
# 如果root结点为空
if not root:
return []
# 结果列表
rst = []
# 双端队列
q = deque([root])
# 层数,这里只是简单的判断:奇数层、偶数层
k = 1
# 如果队列不为空
while len(q) > 0:
# 当前层的结点结果值
r = []
# 下一岑的双端队列
tmp = deque()
# 如果是奇数层
if k == 1:
# 遍历结点
for node in q:
# 添加当前结点的值
r.append(node.val)
# 先判断左子结点
if node.left:
# 插入到队列左侧
tmp.appendleft(node.left)
# 再判断右子结点
if node.right:
# 插入到队列左侧
tmp.appendleft(node.right)
# 如果为偶数层
else:
# 遍历结点
for node in q:
# 添加当前结点的值
r.append(node.val)
# 先判断右子结点
if node.right:
# 插入到队列左侧
tmp.appendleft(node.right)
# 再判断左子结点
if node.left:
# 插入到队列左侧
tmp.appendleft(node.left)
# 添加到结果列表
rst.append(r)
# 反转层数奇偶性
k ^= 1
# 更新队列
q = tmp
return rst