LeetCode 145 Binary Tree Postorder Traversal

LeetCode 145

Binary Tree Postorder Traversal

• Problem Description：
  用栈实现树的后序遍历
  具体的题目信息：
  https://leetcode.com/problems/binary-tree-postorder-traversal/description/
• Solution:
  树的先序遍历顺序：左子树、右子树、根节点
  与先序遍历不同，后序遍历需要确保左右子树均访问过，以下用双栈法和单栈法做出解答。
  （1）双栈法：用一个栈接收节点，一个栈记录该节点是否被重复访问过。
  接收栈第一次访问根节点时，将0标记push进栈，在访问右子树过程中完成第二次访问根节点，此时将0标记换成1标记，表示当该根节点的右子树输出完毕后该节点可以直接输出。

/**
 * 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> treeresult;
        stack<TreeNode*> treenode;
        stack<int> treenum;
        while (!treenode.empty() || root != NULL) {
            while (root) {
                treenode.push(root);
                treenum.push(0);
                root = root->left;
            }
            while (!treenode.empty() && treenum.top()) {
                treenum.pop();
                treeresult.push_back(treenode.top()->val);
                treenode.pop();
            }
            if (!treenode.empty()) {
                treenum.pop();
                treenum.push(1);
                root = treenode.top()->right;
            }
        }
        return treeresult;
    }
};

（2）单栈法：用一个栈接收节点，用一个中间变量prev标记上一个已访问过的节点。若当前访问节点的右子树已经被访问过，则不需要对其进行二次访问。

/**
 * 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> treeresult;
        stack<TreeNode*> treenode;
        TreeNode* prev = NULL;
        while (root != NULL || !treenode.empty()) {
            while (root) {
                treenode.push(root);
                root = root->left;
            }
            if (!treenode.empty()) {
                TreeNode* temp = treenode.top()->right;
                if (temp == NULL || temp == prev) {
                    prev = treenode.top();
                    treeresult.push_back(treenode.top()->val);
                    treenode.pop();
                } else {
                    root = temp;
                }
            }
        }
        return treeresult;
    }
};