LeetCode-102.二叉树的层序遍历

二叉树#

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

示例 1：

1
输入：root = [3,9,20,null,null,15,7]
2
输出：[[3],[9,20],[15,7]]

示例 2：

1
输入：root = [1]
2
输出：[[1]]

示例 3：

1
输入：root = []
2
输出：[]

BFS?的解法

初识层序遍历，与之前的一直递归不太一样。感觉是个挺好用的模版

思路：用切片来模拟队列

1
/**
2
 * Definition for a binary tree node.
3
 * type TreeNode struct {
4
 *     Val int
5
 *     Left *TreeNode
6
 *     Right *TreeNode
7
 * }
8
 */
9
func levelOrder(root *TreeNode) [][]int {
10
    res := [][]int{}
11
    if root == nil {
12
        return res
13
    }
14

15
    queue := []*TreeNode{root}
16
    for len(queue) > 0 {
17
        size := len(queue)
18
        level := []int{}
19
        for i := 0; i < size; i++ {
20
            node := queue[0]
21
            queue = queue[1:]
22
            level = append(level, node.Val)
23

24
            if node.Left != nil {
25
                queue = append(queue, node.Left)
26
            }
27

28
            if node.Right != nil {
29
                queue = append(queue, node.Right)
30
            }
31
        }
32
        res = append(res, level)
33
    }
34
    return res
35
}