题目概述
- 算法说明
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。 - 测试用例
输入:
{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)
}
}
注意事项
- 使用一个数组记录一层节点,一层遍历完后既可追加完下一层的节点,依次循环即可使用一个数组遍历所有层了。
说明
- 当前使用 go1.15.8
- 参考 牛客网--剑指offer
标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目,例如 jz01 代表牛客网剑指offer中01号题目。
注意!!!
- 笔者最近在学习 golang,因此趁机通过数据结构和算法来进一步熟悉下go语言
- 当前算法主要来源于剑指 offer,后续会进一步补充 LeetCode 上重要算法,以及一些经典算法
- 此处答案仅为参考,不一定是最优解,欢迎感兴趣的读者在评论区提供更优解