题目25:589. 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
进阶: 递归法很简单,你可以使用迭代法完成此题吗?
思路分析
递归
- 跟二叉树的前序遍历相似
- 不同点在于这里不仅仅有两个孩子,而是有多个孩子,按照从左到右的顺序压入即可。
程序代码
方法一:
// https://leetcode.cn/problems/n-ary-tree-preorder-traversal/description/
// 589. N 叉树的前序遍历
#include<iostream>
#include<vector>
#include<stack>
using namespace std;
class Node {
public:
int val;
vector<Node*> children;
Node() {}
Node(int _val) {
val = _val;
}
Node(int _val, vector<Node*> _children) {
val = _val;
children = _children;
}
};
// 使用递归的方法
class Solution {
private:
vector<int> res;
public:
vector<int> travel(Node* root)
{
if(root == nullptr)
{
return res;
}
res.push_back(root->val);
// 按照顺序压入即可
for(int i=0;i<root->children.size();i++)
{
travel(root->children[i]);
}
return res;
}
vector<int> preorder(Node* root) {
res = travel(root);
return res;
}
};
复杂度分析
时间复杂度:O(N),一共有 N个节点,每个节点都需要遍历一次。
空间复杂度:O(N) ,主要是数组的开销。
思路分析
迭代
按照从右到左的顺序压入栈,因为栈是先进后出。
程序代码
// 使用迭代的方法
class Solution {
public:
vector<int> preorder(Node* root) {
vector<int> res;
if(root == nullptr)
{
return res;
}
stack<Node*> sta;
sta.push(root);
while(!sta.empty())
{
Node *cur = sta.top();
res.push_back(cur->val);
sta.pop();
// 注意这里要反序,因为出栈是顺序是 后入先出
for(int i=cur->children.size()-1;i>=0;i--)
{
sta.push(cur->children[i]);
}
}
return res;
}
};
题目感悟
更多相关知识:github.com/pingguo1987…