LeetCode102. 二叉树的层序遍历Golang版
1. 问题描述
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
2. 思路
二叉树的层序遍历,又称广度优先遍历(BFS)
二叉树广度优先遍历,通常使用非递归方式,借助队列来实现
- 主节点入队
- 记录队列的长度,此时队列的长度,就是这一层元素的个数
- 遍历这一层所有元素,追加到数组中,同时把下一层元素全部压入队列
3. 代码
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
func levelOrder(root *TreeNode) [][]int {
var res [][]int
if root == nil {
return nil
}
queue := []*TreeNode{root}
for len(queue) > 0 {
length := len(queue)
var layer []int
for i := 0; i < length; i++ {
if queue[0].Left != nil {
queue = append(queue, queue[0].Left)
}
if queue[0].Right != nil {
queue = append(queue, queue[0].Right)
}
layer = append(layer, queue[0].Val)
queue = queue[1:]
}
res = append(res, layer)
}
return res
}