leetcode之N-ary Tree Level Order Traversal（429）

最新推荐文章于 2023-03-05 21:37:42 发布

原创最新推荐文章于 2023-03-05 21:37:42 发布 · 222 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python #算法

leetcode解题专栏收录该内容

129 篇文章

订阅专栏

本文介绍了一种算法，用于解决N叉树的层序遍历问题，通过队列实现，逐步遍历每一层的节点，并返回每层节点值的列表。适用于树的深度不超过1000，节点总数不超过5000的情况。

题目：

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

例如，给定一个 3叉树 :

返回其层序遍历:

[
     [1],
     [3,2,4],
     [5,6]
]

说明:

树的深度不会超过 1000。
树的节点总数不会超过 5000。

python代码:

"""
# Definition for a Node.
class Node(object):
    def __init__(self, val, children):
        self.val = val
        self.children = children
"""
class Solution(object):
    def levelOrder(self, root):
        res,q = [],[root]
        while any(q):
            res.append([node.val for node in q])
            q = [child for node in q for child in node.children if child ]
        return res