剑指 Offer 32 - I. 从上到下打印二叉树（golang版）

• 题目描述

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7],
    3
   / \
  9  20
    /  \
   15   7
返回：
[3,9,20,15,7]
 
提示：
节点总数 <= 1000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
           

• 思路

  1、需要一个tmp切片( []*TreeNode )保存树的节点信息，在遍历树的时候，将节点的值保存在结果res切片中，将节点的左右子树依次保存在tmp中，遍历tmp取值即可

  2、注意输入的二叉树可能是空树

• 代码

func levelOrder(root *TreeNode) []int {
	//保存结果，最后将该切片返回
	res := make([]int,0)
	//处理可能是空二叉树的情况
	if root == nil {
		return res
	}
	//tmp保存二叉树的节点，先将根节点保存在tmp中
	tmp := make([]*TreeNode,0)
	tmp = append(tmp,root)
	//一直循环，直到tmp为空
	for len(tmp) > 0 {
		len := len(tmp)
		//处理这一层级的二叉树，将该层级节点的所有值保存在res中，所有子节点保存在tmp中
		for _, v := range tmp {
			res = append(res,v.Val)
			if v.Left != nil{
				tmp = append(tmp,v.Left)
			}
			if v.Right != nil{
				tmp = append(tmp,v.Right)
			}
		}
		//删掉已处理的层级的节点值
		tmp = tmp[len:]
	}

	return res
}
           

• 测试代码

package main

import "fmt"

type TreeNode struct {
     Val int
     Left *TreeNode
     Right *TreeNode
}

func main() {
	root := &TreeNode{3,nil,nil}
	root.Left = &TreeNode{9,nil,nil}
	root.Right = &TreeNode{20,nil,nil}
	root.Right.Left = &TreeNode{15,nil,nil}
	root.Right.Right = &TreeNode{7,nil,nil}
	//root := &TreeNode{}
	fmt.Println( levelOrder(root) )
}

func levelOrder(root *TreeNode) []int {
	//保存结果，最后将该切片返回
	res := make([]int,0)
	//处理可能是空二叉树的情况
	if root == nil {
		return res
	}
	//tmp保存二叉树的节点，先将根节点保存在tmp中
	tmp := make([]*TreeNode,0)
	tmp = append(tmp,root)
	//一直循环，直到tmp为空
	for len(tmp) > 0 {
		len := len(tmp)
		//处理这一层级的二叉树，将该层级节点的所有值保存在res中，所有子节点保存在tmp中
		for _, v := range tmp {
			res = append(res,v.Val)
			if v.Left != nil{
				tmp = append(tmp,v.Left)
			}
			if v.Right != nil{
				tmp = append(tmp,v.Right)
			}
		}
		//删掉已处理的层级的节点值
		tmp = tmp[len:]
	}

	return res
}