给你二叉树的根节点 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
解题思路
1.大致过程和二叉树的层序遍历一样,只是加了一个flag变量来作为判断是否需要反转遍历出来的数组顺序
2.根据题意的“先从左往右,再从右往左进行下一层遍历”,因此我们需要将偶数层的遍历结果进行反转,所以初始化flag = false即可
代码
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @return {number[][]}
*/
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;
};