LeetCode 「中等」103.二叉树的锯齿形层序遍历

sankigan2025-3-7算法LeetCode

二叉树的锯齿形层序遍历open in new window

题目描述

给你二叉树的根节点 root ,返回其节点值的 锯齿形层序遍历 。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

示例

输入:root = [3,9,20,null,null,15,7]
输出:[[3],[20,9],[15,7]]

输入:root = [1]
输出:[[1]]

输入:root = []
输出:[]

解答

关键思路

为了实现锯齿形层序遍历,需要完成以下任务:

  1. 层序遍历:按层逐个访问二叉树节点
  2. 交替方向:根据当前层的索引,决定是从左到右还是从右到左遍历
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;
};
更新于 2025/3/7 18:11:36
ON THIS PAGE