剑指Offer32-1. 从上到下打印二叉树Golang版

剑指Offer32-1. 从上到下打印二叉树Golang版

1. 问题描述

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

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

[3,9,20,15,7]
提示：

节点总数 <= 1000

2. 思路

二叉树层序遍历，又称广度优先遍历（BFS），通常借助队列来实现。
把二叉树上的数据，按照从上到下，从左到右的顺序，放到队列里。

3. 代码

一种写法（按层来收集数据）

/**
 * 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 := []*TreeNode{root}
    // fmt.Println(queue[0])

    for len(queue) > 0 {
        length := len(queue) 
        // length控制每一层的个数
        for length > 0 {
            if queue[0].Left != nil {
                queue = append(queue, queue[0].Left)
            }
            if queue[0].Right != nil {
                queue = append(queue, queue[0].Right)
            }
            res = append(res, queue[0].Val)
            queue = queue[1:]
            length--
        }
    }

    return res
}

另一种写法（收集所有数据，并不分层处理）

/**
 * 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 := []*TreeNode{root}
 
    for len(queue) > 0 {
        if queue[0].Left != nil {
            queue = append(queue, queue[0].Left)
        }
        if queue[0].Right != nil {
            queue = append(queue, queue[0].Right)
        }
        res = append(res, queue[0].Val)
        queue = queue[1:]
    }

    return res
}