1.题目
给定一个 N 叉树,返回其节点值的后序遍历。
例如,给定一个
3叉树
:
返回其后序遍历:
[5,6,3,2,4,1]
.
2.code
2.1递归版本
/*
// 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>v;
vector<int> postorder(Node* root) {
vector<int>t;
vector<Node*>::iterator it;
if(root==NULL)
return v;
if(root!=NULL)
{
for(it=(root->children).begin();it!=(root->children).end();it++)
{
t=postorder(*it);
}
v.push_back(root->val);
}
return v;
}
};
2.2非递归版本
/*
// 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>v;
vector<Node*>::iterator it;
stack<Node*>s;
stack<int>s1;
Node *tree;
tree=root;
if(root==NULL)
return v;
s.push(tree);
// s1.push(tree->val);
while(!s.empty())
{
tree=s.top();
s.pop();
s1.push(tree->val);
if((tree->children).size()!=0)
{
for(it=(tree->children).begin();it!=(tree->children).end();it++)
{
s.push(*it);
//s1.push((*it)->val);
}
}
}
while(!s1.empty())
{
v.push_back(s1.top());
s1.pop();
}
return v;
}
};
3.分析
额外借助一个栈,我们按照从左到右的顺序将孩子节点压入栈中,这样每次取出的时候都是按照从右到左的顺序,额外用一个栈来记录元素的弹出顺序也就是遍历顺序,根据栈的特征,就可以得到后序序列。
因为后序序列的就是从左到右遍历孩子节点,然后在访问根节点。
以图中例子:
栈s:1 弹出栈顶元素 此时s1元素为1 1有孩子节点 依次入栈
栈s:将 3 2 4压入栈 弹出栈顶元素,此时s1中元素为1 4 由于4没有孩子节点,不需要额外元素进堆栈
栈s: 3 2 弹出栈顶元素 此时s1中元素为1 4 2 由于2没有孩子节点,不需要额外元素进堆栈
栈s: 3 弹出栈顶元素 此时s1中元素为1 4 2 3 由于3有孩子节点,所以依次将孩子节点入栈s
栈s: 5 6 弹出栈顶元素 此时s1 中元素为1 4 2 3 6 由于6没有孩子节点,不需要额外元素进堆栈
栈s: 5 弹出栈顶元素 此时s1 中元素为1 4 2 3 6 5 由于5没有孩子节点,不需要额外元素进堆栈
然后将栈顶元素依次存入vector