102. 二叉树的层序遍历（JS实现）

1 题目

给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。
示例：
二叉树：[3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层次遍历结果：
[
[3],
[9,20],
[15,7]
]

2 思路

这道题的主要思路是用队列来进行层序遍历，由于需要区分每一层，因此额外使用一个数组来存储下一层的节点

3代码

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[][]}
 */
var levelOrder = function(root) {
    
    const result = [];
    if (!root) return result;

    let quque = [root];
    let temp = [], res = [];   // temp用于存储下一层级的节点，而res用户存储当前层级的值
    let p;
    while(quque.length > 0 || temp.length > 0) {
        if (quque.length === 0) {  // 当队列长度为0，说明当前层级所有节点已经访问过
            result.push(res);    //保存当前层级的值
            quque = temp;    //访问下一层
            temp = [];
            res = [];
        }
        p = quque.shift();
        if (p) {
            res.push(p.val);
            temp.push(p.left);
            temp.push(p.right);
        }
    }

    return result;
};