LeetCode102. 二叉树的层序遍历Golang版

LeetCode102. 二叉树的层序遍历Golang版

1. 问题描述

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

2. 思路

二叉树的层序遍历，又称广度优先遍历（BFS）
二叉树广度优先遍历，通常使用非递归方式，借助队列来实现

1. 主节点入队
2. 记录队列的长度，此时队列的长度，就是这一层元素的个数
3. 遍历这一层所有元素，追加到数组中，同时把下一层元素全部压入队列

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
}