1. 问题描述:
给定一个 N 叉树,返回其节点值的前序遍历 。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]
提示:
N 叉树的高度小于或等于 1000
节点总数在范围 [0, 10^4] 内
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-ary-tree-preorder-traversal
2. 思路分析:
分析题目可以知道我们按照定义递归遍历N叉树即可。
3. 代码如下:
from typing import List
class Node:
def __init__(self, val=None, children=None):
self.val = val
self.children = children
class Solution:
res = None
def dfs(self, root: Node):
if not root: return
self.res.append(root.val)
for c in root.children:
self.dfs(c)
def preorder(self, root: 'Node') -> List[int]:
# 注意在这里初始化节点提交上去才是正确的不能够在全局初始化list变量因为这样会导致出现后面的测试用例用的也是当前的list
self.res = list()
self.dfs(root)
return self.res