LeetCode：429. N 叉树的层序遍历

最新推荐文章于 2023-06-28 14:34:10 发布

Bertil

最新推荐文章于 2023-06-28 14:34:10 发布

阅读量105

点赞数

分类专栏：力扣文章标签： leetcode JavaScript BFS

本文链接：https://blog.csdn.net/Bertil/article/details/124846849

版权

力扣专栏收录该内容

139 篇文章 8 订阅

订阅专栏

该博客主要介绍了如何实现N叉树的层序遍历。通过定义一个递归函数traverse，利用深度优先搜索策略，配合节点的层级信息，将节点值按层次依次存入结果数组中。递归结束的条件是遇到空节点。最后返回结果数组，得到层次遍历的序列。这种方法适用于处理任意叉数的树结构，且高度不超过1000，节点数量在0到10^4之间的树。

摘要由CSDN通过智能技术生成

给定一个 N 叉树，返回其节点值的层序遍历。（即从左到右，逐层遍历）。

树的序列化输入是用层序遍历，每组子节点都由 null 值分隔（参见示例）。

示例 1：

在这里插入图片描述

输入：root = [1,null,3,2,4,null,5,6]
输出：[[1],[3,2,4],[5,6]]

示例 2：
在这里插入图片描述

输入：root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出：[[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]

提示：

树的高度不会超过 1000
树的节点总数在 [0, 10^4] 之间

解题思路

1.定义递归函数 traverse：定义一个 depth 记录层数，如果当前层数对应的 res 有值就直接 push，没有就以数组的形式赋值进去；然后向下递归遍历 children 节点，直到结束
2.递归结束的条件：当前遍历的这个节点是空，直接return即可

代码

/**
 * // Definition for a Node.
 * function Node(val,children) {
 *    this.val = val;
 *    this.children = children;
 * };
 */

/**
 * @param {Node|null} root
 * @return {number[][]}
 */
var levelOrder = function(root) {
    let res = [];
    const traverse = (node, depth) => {
        if (!node) return;
        if (res[depth]) {
            res[depth].push(node.val);
        } else {
            res[depth] = [node.val];
        }
        for (let i = 0; i < node.children.length; i++) {
            traverse(node.children[i], depth + 1);
        }
    }
    traverse(root, 0);
    return res;
};