107. 二叉树的层序遍历 II

2022-08-19 14:45:55 浏览数 (1)

107. 二叉树的层序遍历 II

力扣题目链接[1]

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

示例1:

代码语言:javascript复制
输入:root = [3,9,20,null,null,15,7]
输出:[[15,7],[9,20],[3]]

「提示:」

  • 树中节点数目在范围 [0, 2000]
  • 1000 <= Node.val <= 1000

思路:

本题最终的打印顺序刚好和102题顺序相反,因此不难想到,最后将结果数组反转即可。

BFS reverse

总体思路和二叉树的层序遍历相同,唯一不同的地方在于将返回的结果进行逆转。「102. 二叉树的层序遍历」的题解可以看这里[2]。具体的代码逻辑这里就不赘述了。

代码语言:javascript复制
/**
 * 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 levelOrderBottom = function(root) {
    let result = [];
    if (!root) return result; // 如果二叉树为空,则返回空数组
    let queue = [root]; // 初始化队列
    while(queue.length) {
        let temp = [];
        let cur = [];
        while(queue.length) { // 遍历每一层节点
            const node = queue.shift();
            cur.push(node.val);
            if (node.left) temp.push(node.left);
            if (node.right) temp.push(node.right);
        }
        result.push(cur); // 每一层节点值组成的数组
        queue = temp; // 将下一层节点信息赋值给队列
    }
    return result.reverse();
};

DFS

那么除了使用队列的形式进行广度优先遍历以外,还有没有其他方案呢?其实是有的。这里也可以使用深度优先遍历进行求解。具体思路如下:

与普通的DFS不同,这里DFS的时候,需要传递第二个参数,用来表明当前节点的层级。递归的时候维护一个二维数组,确保每一层的节点都存在于同一个数组中,这样就达到了层序遍历的目的。

由于默认深度优先遍历是先处理上层节点的,而本题的要求是从下往上,因此跟广度优先遍历的最终处理一样,需要逆转结果数组并返回即可。

代码语言:javascript复制
/**
 * @param {TreeNode} root
 * @return {number[][]}
 */
var levelOrderBottom = function(root) {
    let result = [];
    const dep = (node, depth) => {
        if (!node) return;
        result[depth] = result[depth] || [];
        result[depth].push(node.val);
        dep(node.left, depth   1)
        dep(node.right, depth   1)
    }
    dep(root, 0);
    return result.reverse();
};

总结

本题是二叉树的层序遍历的衍生版本。分别使用了BFS和DFS进行求解。

参考资料

[1]

力扣题目链接: https://leetcode-cn.com/problems/binary-tree-level-order-traversal-ii/

[2]

这里: https://juejin.cn/post/7089448779140726791

0 人点赞