LeetCode(145)Binary Tree Postorder Traversal

题目如下:

Given a binary tree, return the postorder traversal of its nodes' values.

   1
    \
     2
    /
   3

return [3,2,1].
Note: Recursive solution is trivial, could you do it iteratively?

分析如下：

这是二叉树的后序遍历，题目要求用非递归。之前的文章写过，依稀有些印象，这次自己写了写，还是没写成功。前序后序中序遍历都会用到栈这个结构，这也是由树这种数据结构的特点决定的。比较特殊的是，后序遍历的顺序是"左子树->右子树->根"，它要求左子树和右子树都在根之前输出。根据这个特点，可以这样来进行，每次记录一个pre和cur，cur表示当前节点，pre表示上次输出的节点(是上次输出的节点，不是上次访问的节点)，如果pre正好是cur的左子树或者右子树，那么就可以输出cur，否则，说明还有cur的左子树或者右子树等待输出，也就是还没到cur输出的时候。这是第一个可以让cur输出的条件。另外可以可以让cur输出的条件是，cur是叶子节。

代码如下:

/**
 * 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> res;
        stack<TreeNode*> node_stack;
        if(root==NULL)
            return res;
        node_stack.push(root);
        TreeNode* cur=root;
        TreeNode* pre=NULL;
        while(!node_stack.empty()){
            cur=node_stack.top();
            if((cur->left==NULL&&cur->right==NULL)||((pre!=NULL)&&(pre==cur->left||pre==cur->right))) {
                res.push_back(cur->val);
                node_stack.pop();
                pre=cur; //注意这里，pre仅仅在输出了cur的时候下才更新。
            }else{
                if(cur->right!=NULL)
                    node_stack.push(cur->right);
                if(cur->left!=NULL)
                    node_stack.push(cur->left);
            }
        }
        return res;
    }
};

另外一个进行后序遍历的方法是，特定一个tag，标记当前节点是第几次出栈，如果是第一次出栈，就不让它输出，而是把它的右子树和左子树依次压栈，如果是第二次出，就把它输出出来，做法同样在 之前的文章中写了。