题目描述

请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。

例如:
给定二叉树: [3,9,20,null,null,15,7],

  1. 3<br /> / \<br /> 9 20<br /> / \<br /> 15 7<br />返回其层次遍历结果:

[
[3],
[20,9],
[15,7]
]

提示:

节点总数 <= 1000

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[][]}
 */
var levelOrder = function(root) {
    let res = [];
    let queue = [root];
    while (queue.length) {
        let temp = [];
        let tempQueue = queue;
        queue = [];
        // 当前层数 = 当前结果集数量
        let deep = res.length;
        // 对当前整一层的节点进行处理
        let len = tempQueue.length;
        for (let i = 0; i < len; i++) {
            // 根据深度判定顺序,往左输出还是往又
            let node = (deep % 2 === 0) ? tempQueue[i] : tempQueue[len - i - 1];
            if (node) {
                temp.push(node.val);
            }
            // 用于真正入队列,这样顺序进去才能保证顺序不被打乱
            let pushNode = tempQueue[i];
            if (pushNode) {
                queue.push(pushNode.left, pushNode.right);
            }
        }
        if (temp.length > 0) res.push(temp);
    }
    return res;
};

简化一下写法

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[][]}
 */
var levelOrder = function(root) {
    let res = [];
    let queue = [root];
    while (queue.length) {
        let temp = [];
        let len = queue.length;
        for (let i = len; i > 0; i--) {
            let node = queue.shift();
            if (node) {
                // 这里直接奇偶判定插前还是插后,决定打印顺序
                if (res.length % 2 > 0) {
                    temp.unshift(node.val);
                } else {
                    temp.push(node.val);
                }
                queue.push(node.left, node.right);
            }
        }
        if (temp.length > 0) res.push(temp);
    }
    return res;
};