题目:
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].
题意:
给定一个二叉树,对其进行后序遍历,即先访问左右子树,再访问根节点。
思路:
如果使用递归的方法,那么很简单,只需要先递归访问左右子树,再访问当前根节点。我们考虑使用非递归的方法。
依旧是使用栈的存储结构。只不过这次需要记录根节点元素在栈顶位置是第一次出现还是第二次出现。如果根节点在栈顶是第一次出现,那么说明其左子树已经访问结束,但是右子树还没访问,如果是第二次出现,说明左右子树都已经访问过了,所以根节点这个时候可以访问了。采用的方法依旧是将扫描到某个节点的时候,将这个节点一直往左下角一直到叶子节点,保存入栈。很明显最左下角的节点left这个时候出现在栈顶,也是其第一次出现在栈顶。此时去查看其是否有右孩子,如果没有右孩子,则可以遍历该栈顶节点,如果有右孩子,那么取其右孩子,并按照之前的方法,将右孩子往左下角方向,取出所有的孩子的左孩子。当右子树都访问完的时候。此时left第二次出现在栈顶,所以可以访问并将其出栈。
以上。
代码如下:
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
struct TreeNodePost {
bool first;
TreeNode* node;
TreeNodePost(TreeNode* n) :first(false), node(n) {}
};
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> result;
TreeNodePost* p = ((root == NULL) ? NULL : new TreeNodePost(root));
stack<TreeNodePost*> s;
while (!s.empty() || p != NULL) {
while (p != NULL) {
s.push(p);
p = (p->node->left == NULL) ? NULL : new TreeNodePost(p->node->left);
}
if (!s.empty()) {
if (s.top()->first || s.top()->node->right == NULL) {
result.push_back(s.top()->node->val);
s.pop();
p = NULL;//其没有右子树或者右子树已经访问过了,所以令p=NULL,代表无需再去访问其右子树。
}
else {
s.top()->first = true;
p = (s.top()->node->right == NULL) ? NULL : new TreeNodePost(s.top()->node->right);
}
}
}
return result;
}
};