给定一个 N 叉树,返回其节点值的前序遍历。
例如,给定一个 3叉树 :
返回其前序遍历: [1,3,5,6,2,4]。
说明: 递归法很简单,你可以使用迭代法完成此题吗?
思路分析: 和二叉树的前序遍历思想一致,先访问根节点,在访问子节点。
请先翻阅 LeetCode 二叉树的前序遍历(递归、递推)
/*
// Definition for a Node.
class Node {
public:
int val;
vector<Node*> children;
Node() {}
Node(int _val, vector<Node*> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
public:
vector<int> result;
vector<int> preorder(Node* root) {
myPreorder(root);
return result;
}
//先序遍历:先根节点,再子节点
void myPreorder(Node* root){
if (root == NULL){
return;
}
result.push_back(root->val);//先根节点
for (auto child : root->children){//再子节点
myPreorder(child);
}
}
};
方法二:递推法。
/*
// Definition for a Node.
class Node {
public:
int val;
vector<Node*> children;
Node() {}
Node(int _val, vector<Node*> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
public:
vector<int> preorder(Node* root) {
vector<int> result;
stack<Node*> myStack;//辅助栈
while (root != NULL || !myStack.empty()) {
if (root == NULL) {
root = myStack.top();
myStack.pop();
}
//先序遍历:先根节点,再子节点
result.push_back(root->val);//先根节点
//注意栈是先进后出,所以需要倒置放入所有子节点
for (auto it = root->children.rbegin(); it != root->children.rend(); ++it) {
myStack.push(*it);
}
root = NULL;
}
return result;
}
};