Leetcode每日一题：589. N 叉树的前序遍历

原创于 2022-03-10 20:00:26 发布 · 560 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #深度优先 #算法 #python

leetcode练习专栏收录该内容

47 篇文章

订阅专栏

本文介绍了如何在n叉树中使用递归和迭代方法实现前序遍历。递归方法通过根节点、子节点遍历，而迭代法利用栈来控制节点顺序。核心在于理解遍历顺序和数据结构的运用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述
前序遍历二叉树的要点就是根左右
在这里遍历的是n叉树，因此先访问根节点，然后再遍历根节点的每个孩子就可以了

递归解法

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

class Solution:
    def preorder(self, root: 'Node') -> List[int]:
        res = []
        def dfs(root):
            if root is None:
                return 
            res.append(root.val)
            for ch in root.children:
                dfs(ch)
        dfs(root)
        return res

迭代解法

采用栈来解决，先访问根节点，然后将根节点的孩子倒序入栈，再判断每个孩子是否是根节点，是的话重复上面的操作


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

class Solution:
    def preorder(self, root: 'Node') -> List[int]:
        stack, ans = [root], []
        while stack:
            node = stack.pop()
            if node:
                ans.append(node.val)
                for child in node.children[::-1]:
                    stack.append(child)
        return ans