这是树的第7篇算法,力扣链接。
给你二叉树的根节点
root
,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。示例 1:
输入:root = [3,9,20,null,null,15,7] 输出:[[3],[20,9],[15,7]]
这道题本质还是基础的层级遍历,不过唯一不同的是需要将奇数位的subQuery翻转一下。
func zigzagLevelOrder(root *TreeNode) [][]int {
var result [][]int
if root == nil {
return result
}
queue, level := []*TreeNode{root}, 0
for len(queue) > 0 {
var newQueue []*TreeNode
var subResult []int
for _, node := range queue {
if node.Left != nil {
newQueue = append(newQueue, node.Left)
}
if node.Right != nil {
newQueue = append(newQueue, node.Right)
}
subResult = append(subResult, node.Val)
}
if level%2 == 1 {
for i, n := 0, len(subResult); i < n/2; i++ {
subResult[i], subResult[n-1-i] = subResult[n-1-i], subResult[i]
}
}
result = append(result, subResult)
level++
queue = newQueue
}
return result
}