树的结构决定了有关他的问题使用递归是最方便的,二叉树来说,父亲,左孩子右孩子,多么清晰明了。但是这也造成了我的思维定式,如果不用递归解决树的问题,我还真不是很会写,即使是最简单的这种遍历问题。
递归转非递归一般得借助栈。后序遍历是先访问左右孩子,再访问根节点,我们在访问左右孩子之前一定是先经过父亲节点了,那么怎么办呢,先把父亲节点放到栈里,每次出栈的时候再访问它。树最左下的叶子节点一定是先被访问的,应该先沿着左孩子这条线往下找到这个节点,然后看看它有没有右孩子,如果有右孩子,说明它还不够左下,从右孩子的左孩子这条线继续找。获得栈顶元素时,不能马上出栈,要看看它有没有右孩子,如果有右孩子,且没有被访问过,那么应当继续从它右孩子那里走,如果访问过了,说明当前节点的左右孩子都访问完毕了,可以将其出栈。这个访没访问过,并不需要单独开一个数组来记录状态,只需要记录上个访问的节点,因为访问过右孩子又回到父亲的这种情况,上一个访问的一定是最近的这个右孩子。一个指针就可以了。
/**
* 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> v;
stack<TreeNode*> s;
TreeNode *pre = NULL;
TreeNode *pNode = root;
while(pNode || !s.empty()){
while(pNode){
s.push(pNode);
pNode = pNode->left;
}
if(!s.empty()){
TreeNode *t = s.top();
if(t->right && t->right != pre){
pNode = t->right;
}else{
v.push_back(t->val);
pre = t;
s.pop();
}
}
}
return v;
}
};