LeetCode-----第145题-----二叉树的后序遍历

二叉树的后序遍历

难度：困难

给定一个二叉树，返回它的 后序 遍历。

示例:

输入: [1,null,2,3]  
   1
    \
     2
    /
   3 

输出: [3,2,1]

进阶: 递归算法很简单，你可以通过迭代算法完成吗？

 

题目分析：

      后序遍历非递归版，必须掌握基础知识。后序遍历是左右根，所以根会遍历两次，我们不能直接输出，这样的话根会重复输出两次。我们申明一个节点作为标志，相当于一个last_pMove，记录移动节点上一个节点，每次输出的时候需要判断右边节点是否为空，或者走过，如果走过那么输出，如果没有走过或者不为空，那么我们先记录该节点，然后往右走，接着走到最左边。以此类推。

参考代码：

/**
 * Definition for a binary tree node.
 * 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;
        if(root == NULL)
            return res;
        TreeNode* pMove = root;
        stack<TreeNode*> node_stack;
        TreeNode* flag_node = NULL;
        //先走到最左边
        while(pMove)
        {
            node_stack.push(pMove);
            pMove = pMove->left;
        }

        while(!node_stack.empty())
        {
            pMove = node_stack.top();
            node_stack.pop();
            //判断右节点为空，或者走过右边了，那么根节点输出
            if(pMove->right == NULL || pMove->right == flag_node)
            {
                res.push_back(pMove->val);
                flag_node = pMove;
            }else{
                //如果右节点还没走过，那么我们往右走并记录
                node_stack.push(pMove);
                pMove = pMove->right;
                //在走到最左边，边走边记录
                while(pMove)
                {
                    node_stack.push(pMove);
                    pMove = pMove->left;
                }
            }
        }
        return res;
    }
};