【LeetCode】103. 二叉树的锯齿形层序遍历

持续创作，加速成长！这是我参与「掘金日新计划 · 6 月更文挑战」的第7天，点击查看活动详情

题目

给你二叉树的根节点 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

题解

思路

这题思路很简单，在层序遍历打印每层结点的基础上，新增一个 flag 变量控制当前层是正序打印还是逆序打印。

算法流程

• 新建 flag 变量(1 正序，-1 逆序)、根据每层结点个数控制循环几次的 queue，要返回的 result，并初始化 queue = [root]
• 通过判断 queue.length 看是否已遍历完
• 在遍历每层时，先确定该层有多少个结点(当前队列长度) len = queue.length，用来控制循环的次数，同时初始化放当前层结点的数组为空 now = []
• 然后循环，每次循环先弹出结点，然后通过 flag 判断压入 now 的方式，正序就 push()，逆序就 unshift()。 如果该结点有左右子树，那就压入队列，以便下一层继续循环
• 遍历完后将装载当前层结点的 now 压入 result 数组即可，同时对 flag 取反，以便下一层换个顺序遍历

代码

/**
 * 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[][]}
 */
var zigzagLevelOrder = function(root) {
    if(!root)
        return [];

    let flag = 1; // 控制当前层数是 顺序 还是 逆序 打印
    let result = [], queue = [root];

    while(queue.length) {
        let len = queue.length; // 当前队列长度，即当前层数的结点个数
        let now = [];
        for(let i = 0; i < len; i++) {
            let node = queue.shift();
            // 我这里认定 flag === 1 时，是顺序打印，-1 是逆序
            if(flag > 0) {
                now.push(node.val); // 顺序是 push
            } else {
                now.unshift(node.val); // 逆序是 unshift
            }
            
            // 当前结点有左右子树，那就压入队列，等待下一层的 for 循环进行遍历
            if(node.left) queue.push(node.left);
            if(node.right) queue.push(node.right);
        }
        result.push(now);
        flag = -flag;
    }

    return result;
};

结语

业精于勤，荒于嬉；行成于思，毁于随。