leetcode 102. 二叉树的层序遍历

102. 二叉树的层序遍历 - 力扣（Leetcode）

使用slice

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func levelOrder(root *TreeNode) [][]int {
    // 层序遍历 使用size 记录每层数组 queue node 队列

    res := make([][]int, 0)
    queue := make([]*TreeNode, 0)

	if root != nil {
        queue = append(queue, root)
	} else {
        return res
    }


    for len(queue) != 0 {
        size := len(queue)
        levels := make([]int, 0) 

        for i:= 0; i < size; i++ {
            node := queue[0]
            queue = queue[1:len(queue)] //切掉元素0

            levels = append(levels, node.Val) // 添加元素
            if node.Left != nil {
                queue = append(queue, node.Left)
            }
            if node.Right != nil  {
                queue = append(queue, node.Right)
            }
        }
        res = append(res, levels)
    }
    return res
}

使用list

func levelOrder(root *TreeNode) [][]int {
    res := [][]int{}
    if root == nil{//防止为空
        return res
    }
    queue := &list.List{}
    queue.PushBack(root)

    for queue.Len() > 0 {
        length := queue.Len()               //保存当前层的长度，然后处理当前层（十分重要，防止添加下层元素影响判断层中元素的个数）     
        var levelQueue []int
        fmt.Println(length)
        for i := 0; i < length; i++ {
            node := queue.Remove(queue.Front()).(*TreeNode)    //出队列
            if node.Left != nil {
                queue.PushBack(node.Left)
            }
            if node.Right != nil {
                queue.PushBack(node.Right)
            }
            levelQueue = append(levelQueue, node.Val)    //将值加入本层切片中
        }
        res = append(res, levelQueue)          //放入结果集
    }

    return res
}