1,题目要求
Given an n-ary tree, return the preorder traversal of its nodes’ values.
计算一颗多叉树的前序遍历。
2,题目思路
跟后序遍历相似,既可以使用递归的方法,也可以使用非递归的方法——使用栈。需要注意的是,使用栈的情况下,和后续遍历有区别:入栈的孩子节点需要从右往左进入栈,这样才能实现前序遍历。
3,程序源码
方法1:(递归)
class Solution {
public:
vector<int> preorder(Node* root) {
if(root != nullptr)
{
res.push_back(root->val);
for(auto n : root->children)
preorder(n);
}
return res;
}
private:
vector<int> res;
};
方法2:(栈)
class Solution {
public:
vector<int> preorder(Node* root) {
if(root == nullptr)
return res;
s_node.push(root);
while(!s_node.empty())
{
Node* tmp = s_node.top();
res.push_back(tmp->val);
s_node.pop();
for(int i = tmp->children.size()-1;i>=0;i--) {
s_node.push(tmp->children[i]);
}
}
return res;
}
private:
vector<int> res {};
stack<Node*> s_node;
};