leetCode之非递归中序遍历、后序遍历二叉树

核心思想其实很简单，用两个栈，一个栈用来存放结点，另一栈存放整数，分0、1、2三个值。因为遍历核心就是3句话：遍历左子树，遍历右子树，操作（读值）。对于中序遍历，0代表要遍历左子树，1代表操作（读值），2代表遍历右子树；对于后续遍历，0代表遍历左子树，1代表遍历右子树，2代表操作（读值）。

中序遍历的代码如下：

/**
 * 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> inorderTraversal(TreeNode* root) {
        stack<TreeNode *> nodes;
        stack<int> levels;
        TreeNode * temp;
        vector<int> ans;
        int tlevel;
        if(root==NULL) return ans;
        nodes.push(root);
        levels.push(0);
        while(!nodes.empty())
        {
            tlevel=levels.top();
            temp=nodes.top();
            if(tlevel==0)
            {
                levels.pop();
                levels.push(1);
                if(temp->left!=NULL)
                {
                    nodes.push(temp->left);
                    levels.push(0);
                }
            }
            else if(tlevel==1)
            {
                levels.pop();
                levels.push(2);
                ans.push_back(temp->val);
                
            }
            else
            {
                nodes.pop();
                levels.pop();
                if(temp->right!=NULL)
                {
                    nodes.push(temp->right);
                    levels.push(0);
                }
            }
        }
        return ans;
    }
};

后序遍历的代码如下：

/**
 * 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) {
        stack<TreeNode *> nodes;
        stack<int> levels;
        TreeNode * temp;
        vector<int> ans;
        int tlevel;
        if(root==NULL) return ans;
        nodes.push(root);
        levels.push(0);
        while(!nodes.empty())
        {
            tlevel=levels.top();
            temp=nodes.top();
            if(tlevel==0)
            {
                levels.pop();
                levels.push(1);
                if(temp->left!=NULL)
                {
                    nodes.push(temp->left);
                    levels.push(0);
                }
            }
            else if(tlevel==1)
            {
                levels.pop();
                levels.push(2);
                if(temp->right!=NULL)
                {
                    nodes.push(temp->right);
                    levels.push(0);
                }
            }
            else
            {
                ans.push_back(temp->val);
                nodes.pop();
                levels.pop();
            }
        }
        return ans;
    }
};