Given a binary tree, return the postorder traversal of its nodes' values.
For example:
Given binary tree {1,#,2,3}
,
1 \ 2 / 3
return [3,2,1]
.
Note: Recursive solution is trivial, could you do it iteratively?
非递归方式实现二叉树的后序遍历。
思路:使用栈辅助实现,首先将根节点入栈,然后以栈是否为空为条件进行循环操作,取栈顶元素:
当栈顶元素为叶子节点(栈顶元素的左右孩子节点都是NULL)或者其左右孩子已经遍历过时(previous等于左孩子节点或者右孩子节点),则出栈并记录到vi数组中,同时使将该节点赋给previous指针,本次遍历的节点作为下一次判断的上一个节点。
如果以上条件都不满足,说明栈顶节点的孩子节点还没有遍历到,那么将左右孩子节点入栈,为保证出栈时左孩子节点在右孩子节点前面,应该先将右孩子入栈。
AC code:
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution
{
public:
vector<int> postorderTraversal(TreeNode *root) //非递归实现后序遍历
{
vector<int> vi;
if(root==NULL)
return vi;
TreeNode *previous=NULL;
stack<TreeNode*> st;
st.push(root);
while(!st.empty())
{
root=st.top();
if((root->left==NULL&&root->right==NULL)||((previous!=NULL&&previous==root->left)||(previous!=NULL&&previous==root->right)))
{
vi.push_back(root->val);
st.pop();
previous=root;
}
else
{
if(root->right!=NULL)
st.push(root->right);
if(root->left!=NULL)
st.push(root->left);
}
}
return vi;
}
};