首先来看看前序/中序/后序遍历的简单流程
前序遍历：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）
扫描二维码关注公众号，回复：
10531566 查看本文章
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;
}
};后序遍历，你会了吗？
