102. Binary Tree Level Order Traversal

Leetcode link

题目简介

/**
 * @param {TreeNode} root
 * @return {number[][]}
 */

题目给我们一个二叉树的根 root，要求我们输出其层序遍历的节点值

每一层的节点都要包裹在一个数组中

解题思路

层序遍历我们一般使用 bfs 的思路

我们维护一个数组 queue 存放每一层的元素

当 queue 长度不为 0 的时候我们可以取出当前所有元素放进数组 layer 中

当 queue 当前所有元素取完之后，我们需要把下一层的元素一次 push 近 queue 中方便下次遍历

针对 root 为 null 的情况我们可以在一开始特殊处理

Javascript

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[][]}
 */
var levelOrder = function(root) {
    if(!root) {
        return []
    }

    const queue = [root]
    const res = []

    while(queue.length > 0) {
        let len = queue.length
        const layer = []
        while(len-- > 0) {
            const node = queue.shift()
            layer.push(node.val)
            if(node.left) {
                queue.push(node.left)
            }
            if(node.right) {
                queue.push(node.right)
            }
        }
        res.push(layer)
    }

    return res
};