JavaScript算法19- 二叉树的层序遍历（leetCode:102）

二叉树的层序遍历

---

一、题目

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

示例

输入：root = [3,9,20,null,null,15,7]
输出：[[3],[9,20],[15,7]]

二、题解

• 层序遍历就是广度优先遍历
• 不过在遍历的时候需要记录当前节点所处的层级，方便将其添加到不同的数组中

/**
 * 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 q = [[root, 0]];
   let res = []
   while (q.length > 0) {
       const [n, level] = q.shift();
       if (res[level]) {
           res[level].push(n.val)
       } else {
           res.push([n.val])
       } 
      
       n.left && q.push([n.left, level + 1])
       n.right && q.push([n.right, level + 1])
   }
   return res
};