【LeetCode】589. N叉树的前序遍历、590. N叉树的后序遍历

题目描述

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

例如，给定一个 3叉树 :

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

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

 

思路

递归法非常简单，root 为空时直接为空，否则循环调用子节点输出val即可；

迭代法需要利用栈，前序遍历时，先将当前元素放入 res，然后从后往前入栈元素，栈不空时再输出元素；

后序遍历时，先将当前元素放入 res，然后从前往后入栈元素，栈不空时再输出元素，最后再将 res 颠倒（可用方法 reverse(res.begin(), res.end());），返回 res 即为结果。

 

代码（前序——迭代）

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector<Node*> children;

    Node() {}

    Node(int _val, vector<Node*> _children) {
        val = _val;
        children = _children;
    }
};
*/
class Solution {
public:
    vector<int> preorder(Node* root) {
        vector<int> res;
        if(root == NULL) return res;
        stack<Node*> s;
        s.push(root);
        while(!s.empty()) {
            Node* node = s.top();
            s.pop();
            res.push_back(node->val);
            for(int i=node->children.size()-1; i>=0; i--) {
                s.push(node->children[i]);
            }
        }
        return res;
    }
};

 

代码（后续——递归）

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector<Node*> children;

    Node() {}

    Node(int _val, vector<Node*> _children) {
        val = _val;
        children = _children;
    }
};
*/
class Solution {
public:
    vector<int> postorder(Node* root) {
        vector<int> res;
        recursion(root, res);
        return res;
    }
    void recursion(Node* root, vector<int>& res){
        if(!root) return;
        for(int i=0; i<root->children.size(); i++){
            recursion(root->children[i],res);
        }
        //前序就是把这句放在for循环前面
        res.push_back(root->val);
    }
};