leetcode —— 589. N叉树的前序遍历 （使用到遍历多叉树的方法）

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

例如，给定一个 3叉树 :

返回其前序遍历: [1,3,5,6,2,4]。
——————————
解法一：通过递归的方法进行求解，时间复杂度为$O(n)$，空间复杂度为$O(n)$,其C++代码如下：

/*
// 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 {
private:
    vector<int> num;
public:
    vector<int> preorder(Node* root) {
        function(root);
        return num;
    }
    void function(Node* root)
    {
        if(root==NULL)
            return;
        num.push_back(root->val);
        for(int i=0;i<root->children.size();++i)
        {
            function(root->children[i]);
        }
    }
};

解法二：使用迭代的方法，我们使用一个栈来帮助我们得到前序遍历，需要保证栈顶的节点就是我们当前遍历到的节点。我们首先把根节点入栈，因为根节点是前序遍历中的第一个节点。随后每次我们从栈顶取出一个节点 u，它是我们当前遍历到的节点，并把 u 的所有子节点逆序推入栈中。例如 u 的子节点从左到右为 v1, v2, v3，那么推入栈的顺序应当为 v3, v2, v1，这样就保证了下一个遍历到的节点（即 u 的第一个子节点 v1）出现在栈顶的位置。

作者：LeetCode
链接：https://leetcode-cn.com/problems/n-ary-tree-preorder-traversal/solution/ncha-shu-de-qian-xu-bian-li-by-leetcode/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

其C++代码如下。时间复杂度为$O(n)$，空间复杂度为$O(n)$。

/*
// 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 {
public:
    vector<int> preorder(Node* root) {
        vector<int> num;
        if(root==NULL)
            return num;
        stack<Node*> stk;
        stk.push(root);
        while(!stk.empty())
        {
            Node* n1 = stk.top();
            num.push_back(n1->val);
            stk.pop();
            for(int i=n1->children.size()-1;i>=0;i--)
            {
                stk.push(n1->children[i]);
            }
        }
        return num;
    }
};