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

• 题目描述

请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。

例如:
给定二叉树: [3,9,20,null,null,15,7],
    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果：
[
  [3],
  [20,9],
  [15,7]
]
 
提示：
节点总数 <= 1000

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

• 思路
  1、跟上两个题差不多，就是处理时多个判断是否倒序输出的判断，对2取余即可，详细看代码注释

• 代码

func levelOrder(root *TreeNode) [][]int {
	//保存最后结果
	res := make([][]int,0)
	//处理空二叉树情况
	if root == nil {
		return res
	}
	//创建临时切片，将根节点存入tmp中
	tmp := make([]*TreeNode,0)
	tmp = append(tmp,root)
	//num对2取余，根据余数判断是否倒叙输出
	num := 0
	for len(tmp) > 0 {
		//求出当前tmp长度，在处理完该层级二叉树所有节点后，将前len长度的tmp删除
		len := len(tmp)
		//保存该层级节点的Val值
		var one = make([]int,0)
		//将该层级节点的子节点存入tmp中
		for _, v := range tmp {
			if v.Left != nil{
				tmp = append(tmp,v.Left)
			}
			if v.Right != nil{
				tmp = append(tmp,v.Right)
			}
		}
		//对2取余，余数为1则倒序输出
		if num % 2 == 1 {
			for i:=len; i>0; i-- {
				one = append(one,tmp[i-1].Val)
			}
		}else{
			for i:=0; i<len; i++ {
				one = append(one,tmp[i].Val)
			}
		}
		num ++
		//删除前len元素，即删除该层级的节点
		tmp = tmp[len:]
		res = append(res, one)
	}
	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.Left.Left = &TreeNode{6,nil,nil}
	root.Left.Right = &TreeNode{10,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 := make([]*TreeNode,0)
	tmp = append(tmp,root)
	//num对2取余，根据余数判断是否倒叙输出
	num := 0
	for len(tmp) > 0 {
		//求出当前tmp长度，在处理完该层级二叉树所有节点后，将前len长度的tmp删除
		len := len(tmp)
		//保存该层级节点的Val值
		var one = make([]int,0)
		//将该层级节点的子节点存入tmp中
		for _, v := range tmp {
			if v.Left != nil{
				tmp = append(tmp,v.Left)
			}
			if v.Right != nil{
				tmp = append(tmp,v.Right)
			}
		}
		//对2取余，余数为1则倒序输出
		if num % 2 == 1 {
			for i:=len; i>0; i-- {
				one = append(one,tmp[i-1].Val)
			}
		}else{
			for i:=0; i<len; i++ {
				one = append(one,tmp[i].Val)
			}
		}
		num ++
		//删除前len元素，即删除该层级的节点
		tmp = tmp[len:]
		res = append(res, one)
	}
	return res
}