【LetMeFly】589.N 叉树的前序遍历：深度优先搜索(DFS)

力扣题目链接：https://leetcode.cn/problems/n-ary-tree-preorder-traversal/

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

像正常的深度优先搜索一样，写一个函数来实现递归操作。这个函数接受一个节点作为参数：

首先将这个节点的值加入答案数组中，接着依次递归遍历每一个子节点。

从根节点开始调用这个函数后，最终返回答案数组即可。

• 时间复杂度$O(N)$，其中$N$是节点个数
• 空间复杂度$O(N)$

AC代码

C++

class Solution {
private:vector<int> ans;void dfs(Node* root) {if (!root) {return;}ans.push_back(root->val);for (Node* nextNode : root->children) {dfs(nextNode);}}
public:vector<int> preorder(Node* root) {dfs(root);return ans;}
};

Python

# from typing import Optional, List# # Definition for a Node.
# class Node:
#     def __init__(self, val=None, children=None):
#         self.val = val
#         self.children = childrenclass Solution:def dfs(self, root: Optional['Node']) -> None:if not root:returnself.ans.append(root.val)for nextChild in root.children:self.dfs(nextChild)def preorder(self, root: Optional['Node']) -> List[int]:self.ans = []self.dfs(root)return self.ans

同步发文于CSDN，原创不易，转载经作者同意后请附上原文链接哦~
Tisfy：https://letmefly.blog.csdn.net/article/details/136149332