Leetcode #145 Binary Tree Postorder Traversal

题目链接：https://leetcode.com/problems/binary-tree-postorder-traversal/

 

(非递归实现)二叉树的后序遍历。

class Solution
{
public:
    vector<int> postorderTraversal(TreeNode *root)
    {
        vector<int> output;
        if(root == NULL)
        {
            return output;
        }

        stack<TreeNode*> nodeStack;
        TreeNode *recentlyVisited = NULL;

        while(root != NULL || !nodeStack.empty())
        {
            if(root != NULL)
            {
                nodeStack.push(root);
                root = root->left; //找到最左边的叶子节点。
            }
            else //左子树为空。
            {
                root = nodeStack.top(); //回退一个节点。
                if(root->right != NULL && root->right != recentlyVisited) //右子树不为空并且未访问过。
                {
                    root = root->right;
                    nodeStack.push(root);
                    root = root->left;
                }
                else //右子树为空或者已访问过。
                {
                    root = nodeStack.top(); //回退一个节点。
                    nodeStack.pop(); //从栈中删除该节点(刚才只回退过，没有删除)。
                    output.push_back(root->val); //输出该节点。
                    recentlyVisited = root; //记录下右子树的访问情况(记录的过程是从下至上的)。
                    root = NULL; //左右子树输出完毕，重置这个节点，在下次进入循环的时候会分配给它栈顶的节点。
                }
            }
        }

        return output;
    }
};