1、简介
我们在递归的时候说到过,一般我们如果想把递归的算法转换至非递归的实现,我们可以自己利用辅助栈来代替系统栈保存一些信息。所以在实现二叉树的三种非递归遍历的时候,我们需要开辟一个辅助栈来保存一些信息。
2、二叉树结点结构
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
3、先序遍历
2.1 实现思路
我们只需要开辟一个栈,先让根节点入栈。只要栈不空,我们循环的访问栈内每一位结点,如果访问的当前结点有孩子,让结点的右孩子左孩子依次入栈即可。(因为栈是先进后出,先序遍历是根左右,所以我们入栈的时候是先让右孩子入栈)。
3.2 代码实现
class Solution {
public:
void preorderTraversal(TreeNode* root) {
stack<TreeNode*> s;
s.push(root);
while(!s.empty()){
TreeNode *temp = s.top();
s.pop();
if(!temp) continue;
cout<<temp->val<<" ";
if(temp->right) s.push(temp->right);
if(temp->left) s.push(temp->left);
}
}
};
4、中序遍历
4.1 实现思路
中序遍历的顺序是左根右,所以我们要先要访问到最左的那个结点(while(t){t= t->left})。在访问最左结点的过程中将路径上的结点入栈。然后当栈不为空的时候循环访问即可。
比如说最左的结点是1号结点,其父节点是2号结点。那么在找1号结点的过程中2号结点必定已入栈并且在1号结点下面。所以访问完1号结点后,接着会访问2号结点,只需在访问完2号结点时,将当前指针指向其右子树就会去访问其右子树。总体下来的顺序就是左根右。
4.2 代码实现
class Solution {
public:
void inorderTraversal(TreeNode* root) {
stack<TreeNode*> s;
TreeNode *t = root;
while(!s.empty() || t){
while(t){ //找到最左的那个结点
s.push(t);
t = t -> left;
}
if(!s.empty()){
TreeNode *temp = s.top();
s.pop();
cout<<temp->val<<" ";
t = temp->right;
}
}
}
};
5、后序遍历
5.1 实现思路
后序遍历的非递归实现相较于先序和中序有点复杂。因为我们需要记录下结点的状态,后序遍历的顺序是左右根,我们在访问完左孩子去访问右孩子的时候,要经过根节点,所以我们要记录一下根结点的状态,如果这个根节点访问过了才输出。
5.2 代码实现
由于要记录每个结点的状态,我们有两种可以实现的方式:第一种是在二叉树结点的结构体内加一个标志位;第二种是我们采用对组的形式,每个结点绑定一个bool值。在这里我们使用对组的方式实现。
class Solution {
public:
void postorderTraversal(TreeNode* root) {
stack<pair<TreeNode*, bool> > s;
s.push(make_pair(root, false)); //根节点入栈
bool visited;
while(!s.empty()){
pair<TreeNode*, bool> p = s.top();
s.pop();
visited = p.second;
if(!p.first){
continue;
}
if(visited){ //如果该节点被访问过才输出
cout<<p.first->val<<" ";
}
else
{
s.push(make_pair(p.first, true));
if(p.first->right) s.push(make_pair(p.first->right, false));
if(p.first->left) s.push(make_pair(p.first->left, false));
}
}
}
};