Leetcode刷题(第102题)——二叉树的层序遍历

一、题目

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

二、示例

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

输入：root = [1]
输出：[[1]]

输入：root = []
输出：[]

三、思路
二叉树的层序遍历采用的方法是：使用到的数据结构是队列。只不过在本题需要加上层号，方便分组。
四、代码

/**
 * 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 result = []
    let arr = []
    if(!root) return result
    result.push([0, root])
    while(result.length) {
        let curArr = result.shift()
        let index = curArr[0]
        let val = curArr[1]
        if(arr[index]) {
            arr[index].push(val.val)    
        }else {
            let n = val.val
            arr[index] = [n]
        }
        if(val.left) result.push([index + 1, val.left])
        if(val.right) result.push([index + 1, val.right])
    }
    return arr
};

五、总结