前端算法(63)

51 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

题目

给你二叉树的根节点 root ,返回其节点值 自底向上的层序遍历 。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)

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

题目解析

思路一

我们先准备两个队列,一个存放当前所有同级父元素,一个存放所有同级子元素,然后先将根元素放入父队列 当两个队列只要有一个不为空时,进入迭代后并进行判断父队列是否为空,如果父队列不为空,执行dequeue出列操作,并将每个父元素的值放入结果数组的第一个子数组中。将父元素的子元素放入子队列。当父队列清空以后,判断子队列是否有值,如果有值,结果数组中unshift一个空数组。并且直接进入下轮循环,下轮循环进来以后,父队列一定是空的,而子队列一定有值,将子队列中的值出列到父队列中,结束本轮循环。

/**
 * @param {TreeNode} root
 * @return {number[][]}
 */
var levelOrderBottom = function (root) {
    let list = []
    if (!root) {
        return list
    }
    let parentQueue = []
    let childrenQueue = []
    list.push([])
    parentQueue.push(root)
    while (parentQueue.length || childrenQueue.length) {
        while (parentQueue.length) {
            let node = parentQueue.shift()
            if (node.left) childrenQueue.push(node.left)
            if (node.right) childrenQueue.push(node.right)
            list[0].push(node.val)
            if (!parentQueue.length && childrenQueue.length) list.unshift([])
            continue;
        }
        while (childrenQueue.length) {
            parentQueue.push(childrenQueue.shift())
        }
    }
    return list
};

思路二

我们这里使用遍历,先判断一下如果数据为空则直接返回空数组,初始队列只有一个根节点,在定义一个存储最终返回的结果的数组,进行循环,如果在q还有值,也就是还没有遍历完的情况下进行执行,并将队列全部出队,在进行循环并弹出头节点, 在最后一个数组终加入新的元素,如果左右节点有值,则加入队列,等到循环完之后输出的 res 是从上到下的

var levelOrderBottom = function (root) {
    if (!root) return []
    const q = [root]
    let res = []
    while (q.length) {
        let len = q.length
        res.push([])
        while (len--) {
            let n = q.shift()
            res[res.length - 1].push(n.val)
            if (n.left) q.push(n.left)
            if (n.right) q.push(n.right)
        }
    }
    return res.reverse()
};