请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
剑指 Offer 32 - I. 从上到下打印二叉树:打印结果直接保存
剑指 Offer 32 - II. 从上到下打印二叉树 II:打印结果分层保存
本题是从上到下打印二叉树的最终篇:根据奇偶层决定打印顺序
func levelOrder(root *TreeNode) [][]int {
// 特殊场景:根节点为nil时,返回空数组
if root == nil {
return [][]int{}
}
// 初始化辅助队列
var queue []*TreeNode = []*TreeNode{root}
// 初始化结果数组
res := make([][]int, 0)
// 初始化层数
height := 1
// queue为空时,退出循环
for len(queue) > 0 {
// 新建本层结果数组
tmp := make([]int, 0)
// 记录当前本层节点数量
size := len(queue)
// 记录本层结果
for i := 0; i < size; i++ {
// 出队
node := queue[0]
queue = queue[1:]
// 根据奇偶层,选择写入顺序
if height%2 == 0 {
// 偶数层:从头部写入
tmp = append([]int{node.Val}, tmp...)
} else {
// 奇数层:从尾部写入
tmp = append(tmp, node.Val)
}
// 写入出队节点子节点
if node.Left != nil {
queue = append(queue, node.Left)
}
if node.Right != nil {
queue = append(queue, node.Right)
}
}
// 层数+1
height++
// 本层结果写入结果数组
res = append(res, tmp)
}
return res
}
KGo笔记
全部评论