菜鸟系列 Golang 实战 Leetcode —— 面试题32 - I. 从上到下打印二叉树

Posted 2021-03-05 i-dandan

面试题32 - I. 从上到下打印二叉树

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

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

    3
   /   9  20
    /     15   7
返回：

[3,9,20,15,7]
 

提示：

节点总数 <= 1000

题解

层次打印，利用切片保存每层的节点即可

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func levelOrder(root *TreeNode) []int {
    var res []int
    if root==nil{
        return res
    }
    var nodes []*TreeNode
    nodes=append(nodes,root)
    for i:=0;i<len(nodes);i++{
        res=append(res,nodes[i].Val)
        if nodes[i].Left!=nil{
            nodes=append(nodes,nodes[i].Left)
        }
        if nodes[i].Right!=nil{
            nodes=append(nodes,nodes[i].Right)
        }
    }
    return res
}