剑指 Offer 32 - II. 从上到下打印二叉树 II JavaScript 实现

问题描述

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],

3

/
9 20
/
15 7
返回其层次遍历结果：

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

代码

var levelOrder = function(root) {
    if (!root) return [];
    let arr = [];//存放树的根节点
    arr.push(root);
    let res = [];//输出结果
    let level = 0;//树的深度
    while(arr.length) {
        res[level] = [];//初始化数组子元素，存储一层的节点
        let levelNum = arr.length;//当前层的节点个数
        while (levelNum--){
            let node = arr.shift();
            res[level].push(node.val);
            node.left && arr.push(node.left);
            node.right && arr.push(node.right);
        }
        level++;
    }
    return res;
};