Leetcode笔记每日一题 429. N 叉树的层序遍历（22.04.08）

Compass_NULL

于 2022-04-09 21:42:52 发布

阅读量254

点赞数

文章标签：数据结构 python leetcode

本文链接：https://blog.csdn.net/weixin_41624993/article/details/124064781

版权

Leetcode笔记每日一题 429. N 叉树的层序遍历（22.04.08）

Leetcode 每日一题 429. N 叉树的层序遍历（22.04.08）

一、题目

给定一个 N 叉树，返回其节点值的层序遍历。（即从左到右，逐层遍历）。
树的序列化输入是用层序遍历，每组子节点都由 null 值分隔（参见示例）。

示例 1：
示例1

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

示例 2：
示例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] 之间

二、解题思路

· 广度优先搜索（BFS）

题目是求返回N叉树其节点值的层序遍历。简单理解就是一层一层把节点值记录下来，最后输出总结果。可以采用BFS算法具体如下

先判断N叉树是否存在；
列表nodes[]来存在根节点root，res[]来记录存放结果随后在广度优先搜索的每一轮每一层遍历；
将列表nodes[]的值放在一个临时列表cnt[]里,然后从左往右读取这一层每一个节点的子节点放进临时列表next[]里，当这一层遍历完的时候，将cnt[]存放进res[]中，next[]作为新的nodes[]继续循环直到遍历结束输出总的结果res[]

三、Python代码

"""
# Definition for a Node.
class Node:
    def __init__(self, val=None, children=None):
        self.val = val
        self.children = children
"""

class Solution:
    def levelOrder(self, root: 'Node') -> List[List[int]]:
        # 判断N叉树是否存在
        if not root:
            return []
        nodes = [root]
        res = []
        while nodes:
            cur = [] # 建立临时列表存放节点值
            next = [] # 建立临时列表存放子节点
            for node in nodes:
                # 将node的值存放进临时列表cur[]
                cur.append(node.val)
                # 把node的子节点存放进临时列表next[]
                for child in node.children:
                    next.append(child)
            # 把next[]当作新的nodes[]
            nodes = next
            # 获取临时列表cur[]里的值
            res.append(cur)
        return res # 返回结果

· 简化版BFS模板

"""
# Definition for a Node.
class Node:
    def __init__(self, val=None, children=None):
        self.val = val
        self.children = children
"""

class Solution:
    def levelOrder(self, root: 'Node') -> List[List[int]]:
        queue, res = [root], []
        while queue and queue[0]:
            res.append([node.val for node in queue])
            queue = [child for node in queue for child in node.children]
        return res