589. N叉树的前序遍历

N叉树的前序遍历

原题：

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

N 叉树 在输入中按层序遍历进行序列化表示，每组子节点由空值 null 分隔（请参见示例）。

进阶：

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

示例 1：

输入：root = [1,null,3,2,4,null,5,6]
输出：[1,3,5,6,2,4]

示例 2：

输入：root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出：[1,2,3,6,7,11,14,4,8,12,5,9,13,10]

提示：

• N 叉树的高度小于或等于 1000
• 节点总数在范围 [0, 10^4] 内

解题方法：

解法一（递归）：

class Solution {

public:
    vector<int> tree;        //在函数外声明作为全局变量，就可以避免放进函数的参数中

    vector<int> preorder(Node* root) {
        if(!root) return {};
        Pre(root);
        return tree;
    }

    void Pre(Node* root)
    {
        if(!root) return;
        tree.push_back(root->val);
        for(int i=0;i<root->children.size();i++)
        {
            Pre(root->children[i]);
        }
    }
};

解法二：

class Solution {
public:
    vector<int> preorder(Node* root) {
        if(!root) return {};
        vector<int> tree;
        stack<Node*> st;   
        Node* node;
        st.push(root);
        while(!st.empty())
        {
            node=st.top();
            st.pop();
            tree.push_back(node->val);
            for(int i=node->children.size()-1;i>=0;i--)  //注意要从右向左放入，才能从左向右取出
            {
                st.push(node->children[i]);
            }
        }
        return tree;
    }
};

做题小结：

针对解法一：

1. 将tree声明为全局变量是一大亮点，这样就避免了在函数中使用引用，降低了代码的复杂度
2. 因为vector作为返回值，递归很麻烦，因此另外声明一个递归，返回void就简单很多了

针对解法二：

1. 需要注意的是，放入栈中时，一定是从右往左的顺序，这样才出栈时才能保证从左往右