【路飞】算法与数据结构-二叉树的锯齿形层序遍历

129 阅读1分钟

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;
};