同一段代码搞定二叉树的前序/中序/后序遍历

首先来看看前序/中序/后序遍历的简单流程

前序遍历：root, left, right;

中序遍历：left, root, right;

后序遍历：left, right, root;

这个前中后，说的是遍历root结点的顺序，左右子树的顺序始终没变。

递归的方法很简单，因此不再赘述。迭代的方法却往往让人“一看就会，一写就废”。

下面介绍一种方法，记住这个套路后，任它哪种“序”都不虚。

核心思想如下：

• 使用颜色标记节点状态，未访问的节点标记为白色，访问过的节点标记为灰色；
• 如果遇到白色节点，将其标为灰色，然后将其（左中右，顺序根据具体遍历形式而定）子节点依次入栈；
• 如果遇到灰色节点，则将此节点的指输出。

前序遍历（实际遍历顺序是root, left, right;那么子节点入栈顺序是right, left, root）

C++代码如下：

class Solution{
    vector<int> ans;
public:
    vector<int> preOrderTraverse(TreeNode* root)
    {
        int white = 0;
        int gray = 1;
        stack<pair(int, TreeNode*)>s;
        s.push(make_pair(white, root));
        while(!s.empty())
        {
            int color = s.pop().first;
            TreeNode* t = s.pop().second;
            if(t == NULL) continue;
            if(color == white)
            {
                //入栈顺序：前序遍历的实际遍历顺序：root,left,right；则入栈顺序相反：right,left,root
                s.push(make_pair(white, t->right));
                s.push(make_pair(white, t->left));
                s.push((make_pair(gray, t)));
            }
            else
            {
                ans.push_back(t->val);
            }
        }
        return ans;
    }
};

中序遍历（实际遍历顺序是left, root, right;那么子节点入栈顺序是right,root, left）

C++代码如下：

class Solution{
vector<int> ans;
public:
    vector<int> InOrderTraverse(TreeNode* root){
        int white = 0;
        int gray = 1;
        stack<pair<int,TreeNode*>>s;
        s.push(make_pair(white,root));
        while(!s.empty())
        {
            int color = s.top().first;
            TreeNode* t = s.top().second;
            s.pop();
            if(t == NULL)
            {
                continue;
            }
            if(color == white)
            {
                //入栈顺序：中序遍历的实际遍历顺序：left,root,right；则入栈顺序相反：right,root,left
                s.push(make_pair(white, t->right));
                s.push(make_pair(gray, t));
                s.push(make_pair(white, t->left));
            }
            else
            {
                ans.push_back(t->val);
            }
        }
    return ans;
    }
};

后序遍历，你会了吗？