题目描述
给定一个 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]
提示:
- 节点总数在范围 [0, 104]内
- 0 <= Node.val <= 104
- n 叉树的高度小于或等于 1000
题解思路
方法一-递归
我们采用递归的方法对N叉树进行遍历然后将节点值存储到数组中
方法二-栈方法
我们遍历根节点 并且将根节点的 子树以倒序的形式存储进栈中
然后依次弹出 直到栈空
题解代码
递归
class Solution:
def preorder(self, root: 'Node') -> List[int]:
ans=[]
def helper(node):
if not node:
return
ans.append(node.val)
for c in node.children:
helper(c)
helper(root)
return ans
栈方法
class Solution:
def preorder(self, root: 'Node') -> List[int]:
if not root:
return []
ans=[]
q=[root]
while q:
node=q.pop()
ans.append(node.val)
q.extend(node.children[::-1])
return ans
- append添加元素是将节点的孩子节点作为一个整体添加到栈里
- extend则是将子节点依次迭代添加进入栈