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;}
};
总结:
这两道题目就是二叉树前后序遍历的衍生,和二叉树的前后序遍历一样的操作,主要还是练习前后序操作的思想。