0589：N 叉树的前序遍历

力扣第 589 题

题目

给定一个 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

进阶：递归法很简单，你可以使用迭代法完成此题吗?

分析

#1

先写出递归算法，显然，将根节点、每个子树的前序遍历拼接起来即可。

1
2
3
4
5
6
7
8

def preorder(self, root: 'Node') -> List[int]:
	if not root:
		return []
	res = [root.val]
	if root.children:
		for child in root.children:
			res.extend(self.preorder(child))
	return res

56 ms

#2

迭代算法和 0144 类似，注意入栈顺序是子树的反序。

解答

1
2
3
4
5
6
7
8
9

def preorder(self, root: 'Node') -> List[int]:
	res, stack = [], [root]
	while stack:
		node = stack.pop()
		if node:
			res.append(node.val)
			if node.children:
				stack.extend(node.children[::-1])
	return res

52 ms