牛客网 NC15 求二叉树的层序遍历 javascript解法

NC15 求二叉树的层序遍历

中等

知识点：树、广度优先搜索(BFS)
描述
给定一个二叉树，返回该二叉树层序遍历的结果，（从左到右，一层一层地遍历）
例如：
给定的二叉树是{3,9,20,#,#,15,7},

该二叉树层序遍历的结果是
[
[3],
[9,20],
[15,7]
]

提示:
0 <= 二叉树的结点数 <= 1500

示例1
输入：
{1,2}
复制
返回值：
[[1],[2]]

思路：

层序遍历，可以说是广度优先遍历，因为是先遍历完每一层，才到下一层，符合队列的先进先出特性，所以可以考虑用队列求解。
关键点：队列的长度是每层的子结点个数

代码如下：

/*
 * function TreeNode(x) {
 *   this.val = x;
 *   this.left = null;
 *   this.right = null;
 * }
 */

/**
  * 
  * @param root TreeNode类 
  * @return int整型二维数组
  */
function levelOrder( root ) {
    // write code here
    if(!root) return [];
    let queue = [root], result = [];
    while(queue.length) {
        let size = queue.length;// 队列此时长度刚好就是子结点数，即循环数
        let arr = [];// 在该层新建一个临时数组储存该层所有结点，在最后再压入要返回的数组
        for(let i = 0; i < size; i++) {
            let x = queue.shift();
            arr.push(x.val);
            if(x.left) queue.push(x.left);
            if(x.right) queue.push(x.right);
        }
        result.push(arr);
    }
    return result;
}
module.exports = {
    levelOrder : levelOrder
};

时间和空间复杂度如下