Leetcode 589.N叉树的前序遍历(N-ary Tree Preorder Traversal)

夙愿已清 提交于 2020-02-27 15:17:59

Leetcode 589.N叉树的前序遍历

1 题目描述(Leetcode题目链接

  给定一个 N 叉树,返回其节点值的前序遍历。例如,给定一个 3叉树 :

在这里插入图片描述

返回其前序遍历: [1,3,5,6,2,4]。
说明: 递归法很简单,你可以使用迭代法完成此题吗?

2 题解

  用栈的数据结构,方法同二叉树的前序遍历

"""
# 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]:
        if not root:
            return []
        retv = []
        stack = [root]
        while stack:
            root = stack.pop()
            retv.append(root.val)
            stack.extend(root.children[::-1])
        return retv
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!