注:本文的代码实现使用的是 JS(JavaScript),为前端中想使用JS练习算法和数据结构的小伙伴提供解题思路。
描述
给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层序遍历如下:
[
[3],
[20,9],
[15,7]
]
解题思路
这道题只需要在层序遍历法的结果的基础上,使用一个布尔类型的变量,判断这一层的节点值是否需要反转即可。每次统计完这层节点值后,将其取反。
当然,你也可以根据所在层数的奇偶进行判断。本人使用的是布尔变量进行的判断。
点击链接即可查看二叉树层序遍历算法的详解 力扣题目——102. 二叉树的层序遍历
层序遍历代码
var levelOrder = function(root) {
// 用于存放最后结果的数组
let result = []
// 初始化队列,若 root 为 null,则队列直接为[], 不进行 while 循环,直接返回result
let queue = root ? [root] : []
// 只要队列中存在元素,则遍历,若不存在元素,说明树已经遍历完毕,退出
while(queue.length !== 0){
// 存放每一层元素
let temp = []
// 遍历 len 个元素,即遍历这一层的所有元素
for(let i = 0, len = queue.length; i < len; i++){
let node = queue.shift()
temp.push(node.val)
// 只有不为空的时候,才需要将子节点 push 到队列中
if(node.left) queue.push(node.left)
if(node.right) queue.push(node.right)
}
result.push(temp)
}
return result
};
这道题目的代码
var zigzagLevelOrder = function(root) {
let queue = root === null ? [] : [root]
let result = []
// 创建布尔变量,以在后面判断是否需要反转某一层的节点值
// 因为第一层不需要反转,第二层需要,所以初始为 false
let isReverse = false
while(queue.length !== 0){
let temp = []
for(let i = 0, len = queue.length; i < len; i ++){
let node = queue.shift()
temp.push(node.val)
if(node.left) queue.push(node.left)
if(node.right) queue.push(node.right)
}
// 根据 isReverse 的值判断是否反转
if(isReverse) temp = temp.reverse()
isReverse = !isReverse
result.push(temp)
}
return result
};