开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第18天,点击查看活动详情
一、题目描述:
给定一个 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
进阶:递归法很简单,你可以使用迭代法完成此题吗?
二、思路分析:
用栈辅助。
首先,将 root 压入栈中
在栈不为空时,对栈进行遍历,每次弹出栈顶元素
若栈顶元素节点不为空,则将该节点值放入结果集中,且将该节点的子节点从右至左压入栈中(这样弹出时就是从左至右,符合前序遍历的顺序)
三、AC 代码:
"""
# 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 = []
stack.append(root)
res = []
while len(stack) > 0:
top = stack.pop()
if top is None:
continue
res.append(top.val)
# 反序插入子节点
children = top.children
for i in range(len(children) - 1, -1, -1):
child = children[i]
stack.append(child)
return res
范文参考:
N叉树的先序遍历在Trie树中应用广泛,用一个成员变量来收取每一个节点。 - N 叉树的前序遍历 - 力扣(LeetCode)