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

朱雀 2020-10-18 00:36 880阅读 0赞

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

例如:  
给定二叉树: \[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
    }

# 提交结果 #

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xobDExMjQyODEwNzI_size_16_color_FFFFFF_t_70][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xobDExMjQyODEwNzI_size_16_color_FFFFFF_t_70]: /images/1602981391451.png