每日一题---剑指 Offer 32 - II. 从上到下打印二叉树 II[力扣][Go]

// 逐层打印
func levelOrder(root *TreeNode) [][]int {
    var sum [][]int
    queue := list.New()
    if root != nil {
        // 将根节点入队
        queue.PushBack(root)
    }
    for queue.Len() != 0  {
        var tem []int
        size := queue.Len()
        // 每层需要出站的个数
        for  ; size >0 ; size-- {
            // 获取头结点
            e := queue.Front()
            tre := e.Value.(*TreeNode)
            tem = append(tem, tre.Val)
            // 移除头结点
            queue.Remove(e)
            // 将节点的左右节点入队
            if tre.Left != nil {
                queue.PushBack(tre.Left)
            }
            if tre.Right != nil {
                queue.PushBack(tre.Right)
            }
        }
        sum = append(sum, tem)
    }
    return sum
}