剑指 32-II. 从上到下打印二叉树II（按行打印）

1. 题目描述

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果：
[
[3],
[9,20],
[15,7]
]

提示：
节点总数 <= 1000

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

2. 题解

1、我的提交

和之前java做题思路一样，这次需要按行打印层次遍历的顺序，因此设置两个变量：

• levelNum: 当前层待打印的节点个数
• nextLevel: 记录下一层待打印节点个数
  当 levelNum==0 时更新变量。

/**
 * 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) {
    const res = []; //存储最终结果
    const queue = []; //借助队列进行层次遍历
    if(root) queue[0] = root;
    let levelNum = 1; //当前层节点个数
    let nextLevel = 0; //下一层节点个数
    let row = []; //存储当前层结果
    while(queue.length > 0){
        let curr = queue.shift();
        row.push(curr.val);
        levelNum--;
        if(curr.left){
            queue.push(curr.left);
            nextLevel++;
        }
        if(curr.right) {
            queue.push(curr.right);
            nextLevel++;
        }
        if(levelNum == 0){
            res.push(row);
            row = [];
            levelNum = nextLevel;
            nextLevel = 0;
        }
    }
    return res;
};

2、更简洁的代码

var levelOrder = function(root) {
    if(!root) return [];
    const queue = [root]; //队列
    const res = []; //存放遍历结果
    let level = 0; //代表当前层数
    while(queue.length){
        res[level] = []; //第level层的遍历结果
        let levelNum = queue.length; //第level层的节点个数
        while(levelNum--){
            const front = queue.shift();
            res[level].push(front.val);
            if(front.left) queue.push(front.left);
            if(front.right) queue.push(front.right);
        }
        level++;
    }
    return res;
};

时间复杂度：O(n);
空间复杂度：O(n)。