首先来看看前序/中序/后序遍历的简单流程
前序遍历: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;
}
};
后序遍历,你会了吗?