0x00 题目
给定一个二叉树,返回其节点值的 锯齿形
层序遍历
即先 从左往右
,再 从右往左
进行下一层遍历
以此类推,层与层之间交替进行
0x01 思路
使用 层序遍历
的方式即可
可以使用一个布尔变量记录方向
来控制 从左往右
,还是 从右往左
0x02 解法
语言:Swift
树节点:TreeNode
public class TreeNode {
public var val: Int
public var left: TreeNode?
public var right: TreeNode?
public init() { self.val = 0; self.left = nil; self.right = nil; }
public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
self.val = val
self.left = left
self.right = right
}
}
解法:
func zigzagLevelOrder(_ root: TreeNode?) -> [[Int]] {
guard let root = root else { return [] }
// 存放结果
var res: [[Int]] = []
// 存放每一层的节点
var queue: [TreeNode] = [root]
// 转换标识
var flag: Bool = false
while !queue.isEmpty {
// 存放下一层的节点
var tmp: [TreeNode] = []
// 存放当前层的节点值
var arr: [Int] = []
while !queue.isEmpty {
let node = queue.removeFirst()
arr.append(node.val)
// 左节点存在
if let left = node.left {
tmp.append(left)
}
// 右节点存在
if let right = node.right {
tmp.append(right)
}
}
// 需要转换
if flag {
arr.reverse()
}
// 切换到下一层
queue = tmp
res.append(arr)
// 标志切换
flag = !flag
}
return res
}