102. 二叉树的层次遍历 go语言实现

题目描述

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

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

    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果：

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

代码实现

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func levelOrder(root *TreeNode) [][]int {
    res := make([][]int, 0)
    if root == nil {
        return res
    }
    queue := make([]map[int]*TreeNode, 0)

    initNode := make(map[int]*TreeNode,0)
    initNode[0] = root
    queue = append(queue, initNode)

    for len(queue) > 0 {
        nodePair := queue[0]
        for key, node := range nodePair {
            if len(res) == key {
                valQueue := make([]int, 0)
                res = append(res, valQueue)
            }
            res[key] = append(res[key], node.Val)
            if node.Left != nil {
                tmpNode := make(map[int]*TreeNode,0)
                tmpNode[key+1]=node.Left
                queue = append(queue,tmpNode)
            }
            if node.Right != nil {
                tmpNode := make(map[int]*TreeNode,0)
                tmpNode[key+1]=node.Right
                queue = append(queue,tmpNode)
            }
        }
        queue= queue[1:]
    }
	
    return res
}

提交结果