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

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

1. 问题描述

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

2. 思路

层序遍历模板题

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 res
    }

    queue := []*TreeNode{root}

    for len(queue) != 0 {
        var layer []int
        length := len(queue)
        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
}