589. N 叉树的前序遍历 (Python 实现)

x33g5p2x  于2022-02-11 转载在 Python  
字(0.6k)|赞(0)|评价(0)|浏览(152)

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

示例 1:

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

示例 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,6,7,11,14,4,8,12,5,9,13,10]

代码:

"""
# 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 []
        temp, res = [root], []
        while temp:
            item = temp.pop()
            res.append(item.val)
            for x in range(len(item.children)-1, -1, -1):
                temp.append(item.children[x])
        return res

相关文章