LeetCode 「中等」103.二叉树的锯齿形层序遍历
题目描述
给你二叉树的根节点 root ,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
示例
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[20,9],[15,7]]
输入:root = [1]
输出:[[1]]
输入:root = []
输出:[]
解答
关键思路
为了实现锯齿形层序遍历,需要完成以下任务:
- 层序遍历:按层逐个访问二叉树节点
- 交替方向:根据当前层的索引,决定是从左到右还是从右到左遍历
Code
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @return {number[][]}
*/
var zigzagLevelOrder = function(root) {
if (!root) return [];
const result = [];
const queue = [root];
while (queue.length) {
const curLevelSize = queue.length;
const curLevel = [];
for (let i = 0; i < curLevelSize; ++i) {
const node = queue.shift();
curLevel.push(node.val);
if (node.left) queue.push(node.left);
if (node.right) queue.push(node.right);
}
// 奇数层正序,偶数层倒序
result.push(result.length % 2 === 1 ? curLevel.reverse() : curLevel);
}
return result;
};
