每日一题---剑指 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
}