LeetCode 589. N叉树的前序遍历

给定一个 N 叉树，返回其节点值的前序遍历。

例如，给定一个 3叉树 :

 

 

 

 

 

返回其前序遍历: [1,3,5,6,2,4]。

 

说明: 递归法很简单，你可以使用迭代法完成此题吗?

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/n-ary-tree-preorder-traversal

递归：

void process(Node* root,vector<int> &ans) {
    ans.push_back(root->val);
    int n = root->children.size();

    for(int i = 0; i < n; i++)
    {
        process(root->children[i],ans);
    }
}
vector<int> preorder(Node* root) {
    vector<int> ans;
    if(root)
        process(root,ans);
    return ans;
}

迭代：

/*
// 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> ans;
        if(!root) return ans;
        stack<Node*> nodeStack;
        nodeStack.push(root);
        while(!nodeStack.empty()) {
            Node* node = nodeStack.top();
            ans.push_back(node->val);
            nodeStack.pop();
            int n = node->children.size();
            for(int i = n-1; i >= 0 ; i--)
            {
                nodeStack.push(node->children[i]);
            }
        }
        return ans;
    }
};