【LeetCode】彩灯装饰记录 III

题目

题目链接

一棵圣诞树记作根节点为 root 的二叉树，节点值为该位置装饰彩灯的颜色编号。请按照如下规则记录彩灯装饰结果：
第一层按照从左到右的顺序记录
除第一层外每一层的记录顺序均与上一层相反。即第一层为从左到右，第二层为从右到左。

思路

层序遍历

解题过程

在层序遍历的过程中收集每层的元素，存入临时数组当一层遍历结束后，根据当前层是奇数还是偶数决定是将临时数组直接推入结果数组还是reverse后再推入。
最后别忘了把临时数组中剩余的元素推入结果数组中。

复杂度

• 时间复杂度: O(n)
• 空间复杂度: O(n)

代码

/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

function decorateRecord(root: TreeNode | null): number[][] {
    if (!root) return [];
    const queue: TreeNodeWithLevel[] = [];
    queue.push(new TreeNodeWithLevel(0, root.val, root.left, root.right));
    const res: number[][] = [];
    let curLevelRes: number[] = [];
    let curLevel = 0;
    while (queue.length) {
        const cur = queue.shift();
        const level = cur.level;
        if (level === curLevel) {
            // 追加
            curLevelRes.push(cur.val);
        } else {
            // 新的一行
            curLevel = level;
            res.push(level % 2 === 1 ? curLevelRes : curLevelRes.reverse());
            curLevelRes = [cur.val];
        }
        if (cur.left) queue.push(new TreeNodeWithLevel(level + 1, cur.left.val, cur.left.left, cur.left.right));
        if (cur.right) queue.push(new TreeNodeWithLevel(level + 1, cur.right.val, cur.right.left, cur.right.right));
        console.log(curLevel, queue.map(item => item.val));
    }
    curLevel++;
    res.push(curLevel % 2 === 1 ? curLevelRes : curLevelRes.reverse());
    return res;
};

class TreeNodeWithLevel extends TreeNode {
    level: number;
    constructor(level: number, ...args) {
        super(...args);
        this.level = level;
    }
}