leetcode-二叉树的前中后遍历的迭代方法

题目就不用列了，递归法是很简单的，但是迭代法还是有一定难度，在此记录

前序（easy）：

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> res;
        if(root == NULL)
            return res;
        stack<TreeNode*> s;
        s.push(root);
        while(!s.empty()){
            root = s.top();
            res.push_back(root->val);
            s.pop();
            if(root->right!=NULL)
                s.push(root->right);
            if(root->left!=NULL)
                s.push(root->left);
        }
        return res;
        
    }
};

中序（medium）：

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        if(root == NULL)
             return res;
        stack<TreeNode*> s;
        
        while(!s.empty() || root!=NULL){
            while(root!=NULL){
                s.push(root);
                root = root->left;
            }
            root = s.top();
            res.push_back(root->val);
            s.pop();
            root = root->right;
        }
        return res;
        
    }
};

后序（hard）：
方法一
为什么要定义一个prev指针呢？这是因为当一个结点的右子树已经访问过了，我们需要一种方法来标记告诉它，不然会再次进入右子树而导致死循环！因此prev记录的是上一步输出的结点！

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> s;
        TreeNode* prev = NULL;
        while(!s.empty()||root!=NULL){
            while(root!=NULL){
                s.push(root);
                root = root->left;
            }
            root = s.top();
            root = root->right;
            if(root == NULL || root == prev){
                res.push_back(s.top()->val);
                root = NULL; // 这里是当root==prev时，需要再将它变回NULL才能满足循环的结束条件（!s.empty()||root!=NULL）！
                prev = s.top();
                s.pop();
            }
            
        }
        return res;
        
    }
};

可以发现，中序和后序的循环条件都是：

 while(!s.empty()||root!=NULL)

这是因为对于中序和后序，当右子树为空（或以及被输出）并且本身也被输出的时候才结束！

后序的方法二：
我们知道前序是根左右，后序是左右根，所以我们可以使用前序的结果来得出后序

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> s1;
        stack<TreeNode*> s2;
        if(root == NULL)
            return res;
        s1.push(root);
        while(!s1.empty()){
            root = s1.top();
            s2.push(root);
            s1.pop();

            if(root->left!=NULL)
                s1.push(root->left);
            if(root->right!=NULL)
                s1.push(root->right);
        }
        while(!s2.empty()){
            res.push_back(s2.top()->val);
            s2.pop();
        }
        return res;
         
    }
};