【算法】N叉树的遍历

3 次阅读 预计阅读时间: 2 分钟


n

N叉树没有严格意义上的中序遍历。中序遍历通常用于二叉树,它是指先访问左子树,然后访问根节点,最后访问右子树。由于N叉树的节点可能有多个子节点,因此没有明确的“左”和“右”子树的概念,所以无法直接套用中序遍历的概念。

nnnn

题目描述

nnnn

给定一个 n 叉树的根节点 root ,返回 其节点值的 后序遍历 。
n 叉树 在输入中按层序遍历进行序列化表示,每组子节点由空值 null 分隔(请参见示例)。

nnnn

示例 1:

nnnn
20240403160333823-image
nnnn

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

nnnn

示例 2:

nnnn
20240403160403326-image
nnnn

输入: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]n输出:[2,6,14,11,7,3,12,8,4,13,9,10,5,1]

nnnn

代码

nnnn
"""n# Definition for a Node.nclass Node:n    def __init__(self, val=None, children=None):n        self.val = valn        self.children = childrenn"""nn# 先序遍历  根节点->子节点nclass Solution:n    def preorder(self, root: 'Node') -> List[int]:n        def dfs(node):n            if not node: returnn            yield node.valn            for child in node.children:n                yield from dfs(child)n        return [v for v in dfs(root)]nn# 后序遍历  子节点->根节点nclass Solution:n    def postorder(self, root: 'Node') -> List[int]:n        def dfs(node):n            if not node: returnn            for n in node.children:n                yield from dfs(n)n            yield node.valn        return [v for v in dfs(root)]
nnnn

yield 是一个关键字,通常用于 Python 中的生成器函数。它允许函数暂停执行并返回一个中间值,稍后可以继续执行。
每次调用生成器函数时,它会从上一次暂停的地方继续执行,并返回生成器中的下一个值。
这使得生成器可以有效地处理大量数据或者无限序列,而不会一次性将所有数据存储在内存中。

nnnn

附原题

nnnnn
最后更新于 2024-04-03