Leetcode 589.N叉树的前序遍历（简单）递归、非递归实现

题目描述：

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

例如，给定一个 3叉树 :

 

 

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

 

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

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

思路：1.非递归，用单独的栈来记录子树的情况，类比二叉树的非递归遍历可以实现。注意先右子树入栈再左子树。

           2.递归

/*
// 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> result;
        stack<Node*> stk; //stack的类型是指向Node的指针，开始写stack<Node>错
        if(root == NULL) return result;
        stk.push(root);
        while(!stk.empty())
        {
            Node* tmp= stk.top();
            result.push_back(tmp->val);
            stk.pop();
            for(int i=tmp->children.size()-1; i>=0; --i)
            {
                stk.push(tmp->children[i]);
            }
        }
        return result;
    }
};

/*
// 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> result;
    vector<int> preorder(Node* root) {    
        if(root != NULL){
            result.push_back(root->val);//先序根节点，然后递归遍历每一个子树
            for(auto child : root->children){
                preorder(child);
            }
        }
        return result;
    }
};

/*
// 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:
    void preord(Node *root, vector<int>& t)
    {
        if(root == NULL)  return;
        t.push_back(root->val);
        for(auto p: root->children)
            preord(p,t);
    }
    vector<int> preorder(Node* root) {
        vector<int> result;    
        preord(root,result);
        return result;
    }
};