二叉树的锯齿形层序遍历

122 阅读1分钟

二叉树的锯齿形层序遍历

给你二叉树的根节点 root ,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

  示例 1:

plum.jpg

输入: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

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/bi… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路

这题与从上到下打印二叉树 II - 掘金 (juejin.cn)差不多,判断双数的层数就翻转即可

代码

var zigzagLevelOrder = function (root) {
    let stack = [];
    reverse(root, 0, stack);
    return stack;
};

function reverse(root, k, stack) {
    if (!root) return;
    if (!stack[k]) stack[k] = [];
    // 这行代码是翻转关键
    if (k % 2 == 1) stack[k].unshift(root.val);
    else stack[k].push(root.val);
    reverse(root.left, k + 1, stack);
    reverse(root.right, k + 1, stack);
    return;
}