首发于 语雀文档@blueju

    代码:

    1. /**
    2. * Definition for a binary tree node.
    3. * class TreeNode {
    4. * val: number
    5. * left: TreeNode | null
    6. * right: TreeNode | null
    7. * constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
    8. * this.val = (val===undefined ? 0 : val)
    9. * this.left = (left===undefined ? null : left)
    10. * this.right = (right===undefined ? null : right)
    11. * }
    12. * }
    13. */
    14. function levelOrder(root: TreeNode | null): number[][] {
    15. if (root === null) return []
    16. /** 队列 */
    17. let queue: TreeNode[] = []
    18. /** 层序遍历 */
    19. let levelOrder: number[][] = []
    20. /** 添加第一层 */
    21. queue.push(root)
    22. levelOrder.push([root.val])
    23. /** 当队列不为空时 */
    24. while (queue.length !== 0) {
    25. /** 记录当前队列长度 */
    26. const currentLevelQueueLength = queue.length
    27. /** 当前层队列 */
    28. const currentLevelOrder: number[] = []
    29. /** 遍历当前层节点,以此将节点出列,并将下层节点入列 */
    30. for (let i = 0; i < currentLevelQueueLength; i++) {
    31. const outQueueTreeNode = queue.shift()
    32. if (outQueueTreeNode.left) {
    33. queue.push(outQueueTreeNode.left)
    34. currentLevelOrder.push(outQueueTreeNode.left.val)
    35. }
    36. if (outQueueTreeNode.right) {
    37. queue.push(outQueueTreeNode.right)
    38. currentLevelOrder.push(outQueueTreeNode.right.val)
    39. }
    40. }
    41. /** 将当前节点队列加入层序遍历 */
    42. currentLevelOrder.length && levelOrder.push(currentLevelOrder)
    43. }
    44. return levelOrder
    45. };