携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第14天,点击查看活动详情
一、题目
LeetCode N 叉树的前序遍历
给定一个 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
二、题解
需要对给定的树进行前序遍历,而树节点是不定的,前序遍历就是先遍历二叉树的根节点,然后从左到右如果树节点不为空,就依次遍历树节点,对于树节点也是如此。
方法一
对此可以通过使用DFS深度优先搜索,从树的根节点开始遍历,然后对子树进行递归遍历。具体的定义dfs递归函数,参数需要一个root树的根节点来遍历,然后还需要一个记录遍历结果的数组,递归内判断如果当前节点为空的话就结束遍历,说明之后无节点不需遍历了,否则先遍历根节点,将根节点的值加入到结果数组中,然后获取当前节点的子树,子树是不定的为一个数组,我们需要前序遍历从树的左边到右遍历,正好循环遍历子树数组,然后拿子树的根节点去递归遍历,最后遍历结束的时候返回记录的数组即可。
方法二
或者也可以通过BFS广度优先搜索遍历N叉树节点,先将根节点加入队列中,然后循环遍历取出队列元素节点,遍历当前节点,将当前节点值加入到遍历结果集合中,然后遍历子树节点,将子树节点也加入到队列中,因为是前序遍历,所以需要依次按遍历的顺序来加入到队列中,这样就方便继续遍历其他节点了。
三、代码
方法一 Java代码
class Solution {
public List<Integer> preorder(Node root) {
List<Integer> list = new ArrayList<>();
dfs(root, list);
return list;
}
public void dfs(Node root, List<Integer> list) {
if (root == null) {
return;
}
list.add(root.val);
for (Node node : root.children) {
dfs(node, list);
}
}
}
时间复杂度:O(n),需要遍历一次树的所有节点。
空间复杂度:O(n),需要使用数组记录遍历结果以及递归消耗的栈空间。