Pengjielee's blog

题目

给你一个二叉树，请你返回其按 层序遍历 得到的节点值。（即逐层地，从左到右访问所有节点）。

示例：二叉树：[3,9,20,null,null,15,7],

返回其层序遍历结果：

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

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/binary-tree-level-order-traversal 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

JS实现

/**
 * 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 result = [];

  if (root === null) {
    return result;
  }

  // 声明一个队列，队列特点：先进先出
  let q = [];
  // 入队根节点
  q.push(root);

  while (q.length > 0) {
    // 层次数组
    const level = [];
    const len = q.length;

    // 遍历队列
    for (let i = 0; i < len; i++) {
      // 出队，从队头取出元素
      const node = q.shift();
      // 保存节点值 到 层次数组
      level.push(node.val);

      // 左节点入队
      if (node.left != null) {
        q.push(node.left);
      }
      // 右节点入队
      if (node.right != null) {
        q.push(node.right);
      }
    }

    // 保存 每层数组 到结果
    result.push(level);
  }

  return result;
};