力扣题解-589. N叉树的前序遍历（递归和迭代）

题目：589. N叉树的前序遍历

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

例如，给定一个 3叉树 :

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/n-ary-tree-preorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

递归代码

/*
// Definition for a Node.
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 {
    vector<int> ans;
public:
    vector<int> preorder(Node* root) {
        if (!root) {
            return ans;
        }
        ans.push_back(root->val);
        for (auto &node: root->children) {
            preorder(node);
        }
        return ans;
    }
};

迭代代码

class Solution {
public:
    vector<int> preorder(Node* root) {
        vector<int> ans;
        if (!root) {
            return ans;
        }

        stack<Node*> stk;
        stk.push(root);
        while (!stk.empty()) {
            Node* node = stk.top();
            stk.pop();
            ans.push_back(node->val);
            vector<Node*>::reverse_iterator iter = (node->children).rbegin();
            while(iter != (node->children).rend()) {
                stk.push(*iter);
                iter++;
            }
        }
        return ans;
    }
};