二叉树的层序遍历（JS版）

* 二叉树构造函数

/**
 * 树节点
 * @class
 */
function TreeNode(val = 0, left = null, right = null) {
    this.val = val;
    this.left = left;
    this.right = right;
}

 

直接上代码（含必要注释）

1. BFS（广度优先搜索）

/**
 * @param {TreeNode} root
 * @return {number[][]}
 */
var levelOrder = function(root) {
    // 处理根节点为空的特殊情况
    if (root === null) return [];
    // 初始化一个队列（根节点入队）
    const q = [root];
    // 初始化结果数组
    const ans = [];
    // 遍历节点队列
    while (q.length > 0) {
        // 记录当前队列中的节点数量
        let cnt = q.length;
        // 初始化当前层节点的容器
        const level = [];
        // 遍历当前层的节点
        while (cnt-- > 0) {
            const node = q.shift();
            level.push(node.val);
            // 左子节点先入队
            if (node.left !== null) q.push(node.left);
            // 右子节点后入队（以保持从左到右的顺序）
            if (node.right !== null) q.push(node.right);
        }
        // 将当前层的遍历结果推入结果数组
        ans.push(level);
    }

    return ans;
};

 

2. DFS（深度优先搜索）

2.1. 递归版本

var levelOrder = function(root) {
    // 容器（二维数组）
    const ans = [];
    // 搜索所有节点（从第一层开始，对应索引为0）
    dfs(root, 0, ans);

    return ans;
};

var dfs = function(node, level, cont) {
    // 如果节点为空，终止当前搜索
    if (node === null) return;
    // 比对当前结果容器中的层级结果容器数量与当前层（位置），
    // 如果未初始化当前层的容器，则推入一个新数组
    if (cont.length <= level) cont.push([]);
    // 向当前层的结果容器中加入目标值
    cont[level].push(node.val);
    // 递归左子树
    dfs(node.left, level + 1, cont);
    // 递归右子树
    dfs(node.right, level + 1, cont);
};

2.2. 迭代版本

var levelOrder = function(root) {
    // 根节点为空直接返回
    if (root === null) return [];
    // 遍历结果
    const ans = [];
    // 节点栈（模拟系统栈，初始化时，root节点随即入栈）
    const stk = [root];
    // 层级栈（用于记录对应节点的所属层级，root所在层级记为0，也可以为1）
    const level_stk = [0];

    while (stk.length > 0) {
        const node = stk.pop();
        const level = level_stk.pop();
        // 如果当前层级的数据容器在ans中还不存在（初始化一个新数组并推入）
        if (ans.length <= level) ans.push([]);
        // 在当前节点所属层级的数组中记录自身值
        ans[level].push(node.val);
        // 处理当前节点的右子树
        if (node.right !== null) {
            stk.push(node.right);
            level_stk.push(level + 1);
        }
        // 处理当前节点的左子树
        if (node.left !== null) {
             stk.push(node.left);
             level_stk.push(level + 1);
        }
    }

    return ans;
};

 

End