LeetCode:地址
题目要求
给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如: 给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层序遍历如下:
[
[3],
[20,9],
[15,7]
]
思路
这题和二叉树按层打印的思路一样,就是多了几行代码。
需要定义一个是否反转的标志位flag,每隔一层,反转一次即可。
代码
const zigzagLevelOrder = root => {
if (!root) return [];
const res = [];
// 根节点入队
const q = [root];
// 是否需要反转的标志位
let flag = false;
// 当队列还有值时,一直执行
while (q.length) {
let len = q.length;
res.push([]);
while (len--) {
// 获取根节点,根节点出队
const n = q.shift();
// 根节点加入res栈顶元素
res[res.length - 1].push(n.val);
// 队头左右节点入队
n.left && q.push(n.left);
n.right && q.push(n.right);
}
// 每隔一层,反转一次
flag && res[res.length - 1].reverse();
flag = !flag;
}
return res;
};