[LeetCode] 145. Binary Tree Postorder Traversal

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

Description

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].

Note: Recursive solution is trivial, could you do it iteratively?

解题思路

递归的方式很简单，但是这道题要求用非递归的方式，那么自然想到的就是用栈来解决。由于后序遍历的顺序是 左节点->右节点->根，所以可以先一直沿着左节点往下直到NULL，将这条路径上的所有节点依次放入栈(nStack)中。同时用另一个栈(vStack)，为每一个节点保存状态值，用于弹出栈顶时以辨别是否遍历过其右节点，初始值为false。若已经遍历过其右节点，则同时弹出两个栈的栈顶，并将该节点的值存入返回数组中；若未遍历其右节点，则暂时不弹出，修改 vStack 的栈顶为true，表示已访问过其右节点，并与之前一样，此时的根节点应为该节点的右节点，沿着新的根节点的左节点一直遍历至NULL，重复该过程。

Code

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> nStack;
        stack<bool> vStack;

        while (root != NULL) {
            nStack.push(root);
            vStack.push(false);
            root = root->left;
        }

        while (!nStack.empty()) {
            root = nStack.top();

            if (vStack.top()) {
                res.push_back(root->val);
                nStack.pop();
                vStack.pop();
            } else {
                vStack.top() = true;
                root = root->right;

                while (root != NULL) {
                    nStack.push(root);
                    vStack.push(false);
                    root = root->left;
                }
            }
        }
        return res;
    }
};