102. 二叉树的层序遍历

作者:Annan

更新于:1 分钟前

字数统计:91 字

阅读时长:1 分钟

阅读量:

思路就是用BFS(广度优先遍历),用嵌套数组携带层级信息用于判断是哪一层

var levelOrder = function (root) {
  if(!root) return []
  const res = []
  const queue = [[root, 0]]
  while(queue.length) {
    const [ node, level ] = queue.pop()
    if(res[level]) {
      res[level].push(node.val)
    } else {
      res[level] = [node.val]
    }
    if(node.right) queue.push([node.right, level + 1])
    if(node.left) queue.push([node.left, level + 1])
  }
  return res
};

Contributors

Annan