当前位置：首页 > news >正文

代码随想录训练营第二十一天 |589.N叉数的前序遍历 590.N叉树的后序遍历

news 2025/5/14 8:16:43

589.N叉数的前序遍历：

状态：已做出

思路：

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) {stack<Node*>st;//使用栈来实现前序遍历vector<int>result;if(!root) return result;st.push(root);while(!st.empty()) {Node* temp=st.top();//取出栈顶元素，这就是前序遍历的根节点处理st.pop();result.push_back(temp->val);//以下循环就是让每个节点的子树全部都依次入栈for(int i=temp->children.size()-1;i>=0;--i) {st.push(temp->children[i]);}}return result;}
};

590.N叉树的后序遍历：

状态：已做出

思路：

和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> postorder(Node* root) {stack<Node*>st;vector<int>result;if(!root) return result;st.push(root);while(!st.empty()) {Node* temp=st.top();st.pop();result.push_back(temp->val);//这里的循环和前序遍历相反，这样才能让最后反转的数组符合后序遍历for(int i=0;i<temp->children.size();++i) {st.push(temp->children[i]);}}reverse(result.begin(), result.end());//反转数组return result;}
};