题解 | #二叉树的后序遍历# （非递归）

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 使用stack栈，后序遍历由于最后访问根节点，不能在栈顶拿出节点之后，直接访问节点数据存入vector，而应保证根节点的左子树和右子树中的所有节点都已被访问过后，再去访问根节点。

为了判断是否对某个栈顶节点进行访问，引入一个新变量prev，记录前一个被访问到的节点，若满足以下两种情况之一，则对该栈顶节点进行访问：

1. 该节点的右子节点为nullptr；
2. 该节点的右子节点与之前访问的节点prev相同；
如果不满足条件，那么先去访问栈顶节点的右子树，之后再访问这个栈顶节点。

 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return int整型vector
     */
    vector<int> postorderTraversal(TreeNode* root) {
        // write code here
        //return PostOrderTraversalRecursive(root);
        return PostOrderTraversalNotRecursive(root);
    }

    vector<int> PostOrderTraversalRecursive(TreeNode* root) {
      vector<int> vi;
      if (root == nullptr)
        return vi;

      TraversalRecursive(root, vi);
      return vi; 
    }

    void TraversalRecursive(TreeNode* root, vector<int>& vi) {
      if (root->left != nullptr)
        TraversalRecursive(root->left, vi);

      if (root->right != nullptr)
        TraversalRecursive(root->right, vi);

      if (root != nullptr)
        vi.push_back(root->val);
    }

    vector<int> PostOrderTraversalNotRecursive(TreeNode* root) {
      std::vector<int> result;
      std::stack<TreeNode*> st;
      auto cur = root;
      decltype(root) prev = nullptr;

      while (cur || !st.empty()) {
        while (cur) {
          st.push(cur);
          cur = cur->left;
        }

        auto top = st.top();
        if (top->right == nullptr || top->right == prev) {
          // 访问节点
          result.push_back(top->val);
          st.pop();
          prev = top;
        } else {
          cur = top->right;
        }
      }

      return result;
    }
};