题目描述

leetcode链接：leetcode-cn.com/problems/bi…

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

示例：

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

    3
   / \
  9  20
    /  \
   15   7

返回其层序遍历结果：

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

思路描述

利用广度优先搜索。利用队列的先进先出的特性，先将根节点放入队列，然后遍历当前队列，将根节点弹出，将根节点的左节点和右节点加进队列里。进行第二轮循环，这时是第二层了，再遍历当前队列，将遍历的当前节点弹出队列，同时将其子节点加进队列里，成为第三层，以此类推。最后，等到队列里没有数据了则返回最终结果。

AC代码

/**
 * 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) {
  let ret = [];
  if (!root) {
      return ret;
  }
  let arr = [];
  arr.push(root);
  while(arr.length) {
      const len = arr.length;
      const nodeArr = [];
      for (let i = 0; i < len; i++) {
          const node = arr.shift();
          nodeArr.push(node.val);
          if (node.left) arr.push(node.left)
          if (node.right) arr.push(node.right)
      }
      ret.push(nodeArr)
  }
  return ret;
};

本文正在参与「掘金 2021 春招闯关活动」, 点击查看 活动详情