【每日一题】LeetCode 102. 二叉树的层序遍历 TypeScript

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

示例 1:

复制代码
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]

示例 2:

复制代码
输入:root = [1]
输出:[[1]]

示例 3:

复制代码
输入:root = []
输出:[]

提示:

  • 树中节点数目在范围 [0, 2000]
  • -1000 <= Node.val <= 1000

TypeScript 复制代码
/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

/**
 * Definition for a binary tree node.
 */

function levelOrder(root: TreeNode | null): number[][] {
    // 边界处理:空树返回空数组
    if (!root) return [];

    const result: number[][] = [];
    const queue: TreeNode[] = [root];

    while (queue.length > 0) {
        // 当前层的节点数量(重要:需在入队前固定)
        const levelSize = queue.length;
        const currentLevel: number[] = [];

        // 逐层处理
        for (let i = 0; i < levelSize; i++) {
            // 取出队首节点(shift 时间复杂度 O(n),可用下标优化,但此处清晰为主)
            const node = queue.shift()!;
            currentLevel.push(node.val);

            // 左右子节点入队(为下一层做准备)
            if (node.left) queue.push(node.left);
            if (node.right) queue.push(node.right);
        }

        // 将当前层结果加入总结果集
        result.push(currentLevel);
    }

    return result;
}

共勉