LeetCode102. 二叉树的层序遍历

题目描述

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

/**
 * 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)
 * }
 */
 
/**
解题步骤：
1，广度优先遍历二叉树
2，遍历过程中，记录每个节点的层级，并将其添加到不同的数组中 
3，赶紧关注我的微信公众号获取更多内容吧：手摸手前端进阶   
*/

/**
 * @param {TreeNode} root
 * @return {number[][]}
 */
var levelOrder = function(root) {
  if(!root) return []
  const q = [[root, 0]] 
  const res = []
  while(q.length) {
    const [n, level] = q.shift() // 根节点出队
    if(!res[level]) {
      res.push([n.val])
    } else {
      res[level].push(n.val)
    }
    if(n.left) q.push([n.left, level+1])
    if(n.right) q.push([n.right,level+1])
  } 
  return res
};

// 时间复杂度：O(n)
// 空间复杂度：O(n)

// 题目转载自力扣官网：
// https://leetcode-cn.com/problems/binary-tree-level-order-traversal/
复制代码