树11--按之字形顺序打印二叉树

题目概述

• 算法说明
  请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。
• 测试用例
  输入：
  {8,6,10,5,7,9,11}
  输出：
  [[8],[10,6],[5,7,9,11]]

解析&参考答案

• 解析
  使用链表存放一层节点，并使用一个row来记录奇偶行，偶行从左到右输出，奇行从右到做输出。
• 参考答案

vim jz59.go
package main

import "fmt"

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

func Print(pRoot *TreeNode) [][]int {
	if pRoot == nil {
		return [][]int{}
	}
	var ret [][]int
	length := 1
	row := 0
	nodeList := []*TreeNode{pRoot}
	for length > 0 {
		var arr []int
		for i := 0; i < length; i++ {
			if row%2 == 0 {
				arr = append(arr, nodeList[i].Val)
			} else {
				arr = append(arr, nodeList[length-i-1].Val)
			}
			if nodeList[i].Left != nil {
				nodeList = append(nodeList, nodeList[i].Left)
			}
			if nodeList[i].Right != nil {
				nodeList = append(nodeList, nodeList[i].Right)
			}
		}
		ret = append(ret, arr)
		nodeList = nodeList[length:]
		length = len(nodeList)
		row++
	}
	return ret
}

func main() {
	root := &TreeNode{10, &TreeNode{5, &TreeNode{4, nil, nil}, &TreeNode{7, nil, nil}},
		&TreeNode{12, nil, nil}}
	result := Print(root)
	for _, v := range result {
		fmt.Println(v)
	}
}

注意事项

1. 使用一个数组记录一层节点，一层遍历完后既可追加完下一层的节点，依次循环即可使用一个数组遍历所有层了。

说明

1. 当前使用 go1.15.8
2. 参考 牛客网--剑指offer
   标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目，例如 jz01 代表牛客网剑指offer中01号题目。

注意!!!

• 笔者最近在学习 golang，因此趁机通过数据结构和算法来进一步熟悉下go语言
• 当前算法主要来源于剑指 offer，后续会进一步补充 LeetCode 上重要算法，以及一些经典算法
• 此处答案仅为参考，不一定是最优解，欢迎感兴趣的读者在评论区提供更优解