LeetCode 589 N叉树的前序遍历

题目描述

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

n 叉树 在输入中按层序遍历进行序列化表示，每组子节点由空值 null 分隔。

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

题目很好理解，就是简单的二叉树前序遍历换成了N叉树，因此遍历方法也会分为递归和非递归方法。

解法一： 递归

递归解法和二叉树遍历的基本一致，按照中间，从左边到右边的顺序进行遍历，遍历。

代码：

class Solution {
public:
    vector<int> res;

    void dfs(Node* root){
        if(!root) return;
        // 空
        res.emplace_back(root->val);
        // 先加入根节点
        for(auto node:root->children){
            dfs(node);
        }
    }

    vector<int> preorder(Node* root) {
        // 前序遍历  中间 从左边到右边
        dfs(root);
        return res;
    }
};

解法二：非递归

通过一个栈来实现非递归解法，将一个节点入栈后，再将这个节点的children节点从右往左入栈，再将节点出栈并读取。思想和二叉树的一样。

代码：

class Solution {
public:
    vector<int> res;
    vector<int> preorder(Node* root) {
        // 2. 迭代 通过栈实现
        // 和二叉树的非递归前序遍历基本一样
        if(!root) return {};

        stack<Node*> st;
        // 先创建一个栈来存储节点
        st.push(root);

        while(!st.empty()){
            Node* tmp=st.top();
            st.pop();
            // 一层中的节点需要从右往左入栈
            res.emplace_back(tmp->val);
            for(int i=tmp->children.size()-1; i>=0; i--){
                if(tmp->children[i]!=nullptr){
                    // 此处节点存在才入栈
                    st.push(tmp->children[i]);
                }
            }
        }
        return res;
    }
};