I.从上到下打印二叉树

126 字

1 分钟

I.从上到下打印二叉树

2021-08-31

2026-10-12

数据结构与算法

「剑指Offer」Swift

题目#

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

例如:

给定二叉树: [3,9,20,null,null,15,7],

返回：

1
[3,9,20,15,7]

提示：

节点总数 <= 1000

来源：力扣（LeetCode）

题解#

1
func levelOrder(_ root: TreeNode?) -> [Int] {
2
    guard let r = root else { return [] }
3
    var result: [Int] = []
4
    var queue: [TreeNode] = [r]
5

6
    while queue.isEmpty == false {
7
        let first = queue[0]
8
        result.append(first.val)
9

10
        if let left = first.left {
11
            queue.append(left)
12
        }
13

14
        if let right = first.right {
15
            queue.append(right)
16
        }
17

18
        queue.removeFirst()
19
    }
20

21
    return result
22
}