[数据结构与算法]第九题 剑指 Offer 32 - II. 从上到下打印二叉树 II

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

题目描述

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

返回其层次遍历结果：

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

提示

节点总数 <= 1000

思路

BFS

queue数组： 模拟队列
res数组：存放结果
temp数组：存放每一层的节点数据

• root的根结点入队
• 遍历队列 将队列的元素逐一出队 push进temp数组
• 判断是否存在左右子节点 若存在 则将其push 进队列
  注意：
  queue队列长度会发生改变，在做遍历操作时需要更新遍历终点

代码

/**
 * 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) {
    if(!root){
        return [];
    }
    // 根节点入队
    let queue = [root];
    let res = [];
    while(queue.length){
        let temp = [];
        //更新遍历终点值
        let len = queue.length;
        //遍历队列
        for(let i = 0; i <len ; i++){
            //出队
            let n = queue.shift();
            //将出队元素 push至本层的临时数组 temp
            temp.push(n.val);
            //若存在左右节点 将左右节点入队
            if(n.left) queue.push(n.left);
            if(n.right) queue.push(n.right);
        }
        //遍历一层结束 push进结果数组
        res.push(temp);
    }
    return res;

};