Leetcode刷题(第103题)——二叉树的锯齿形层次遍历

x33g5p2x  于2022-03-09 转载在 其他  
字(0.9k)|赞(0)|评价(0)|浏览(223)

一、题目

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

二、示例

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

三、思路
本题在层序遍历的基础上,新增了锯齿形,也就是按照z自行进行遍历。所以我们只需要记录层数的下标即可,在每一次层数为奇数时,则从右向左遍历,当每一层为偶数时,则从左向右遍历。
四、代码

/**
 * 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) {
    let result = []
    let arr = []
    if (!root) return arr
    result.push([0, root])
    while (result.length) {
        let cur = result.shift()
        let index = cur[0]
        let val = cur[1]
        if (arr[index]) {
            if (index % 2) {
                arr[index].unshift(val.val)
            } else {
                arr[index].push(val.val)
            }
        } else {
            arr[index] = [val.val]
        }
        if (val.left) result.push([index + 1, val.left])
        if (val.right) result.push([index + 1, val.right])
    }
    return arr
};

五、总结

相关文章