二叉树的遍历可以使用递归的方式实现,并且代码非常简单。而递归实际就是函数反复的调用本身,在栈上反复压栈。所以我们可以用栈来模拟实现递归。
1.前序遍历
(1)栈是后进先出的特点,所以无条件的把栈的根节点入栈,在把栈顶元素输出之后依次把右孩子,左孩子压入栈中。
代码如下:void _PrevOrder(Node * root)
{
stack s;
if (root == NULL)
{
return;
}
s.push(root);//将第一个元素入栈
while (!s.empty())//当栈不为空时
{
root = s.top();
cout <_data <";//打印节点
s.pop();
//栈的特点,后进先出,所以,先压右子树
if (root->_right)//遍历右子树
{
s.push(root->_right);
}
if (root->_left)//遍历左子树
{
s.push(root->_left);
}
}
}
2.中序遍历
(1)一直入栈,一直到二叉树的最左边最下边的节点。
(2)按照中序遍历的特点:左子树->根节点->右子树,输出栈顶的元素,并且弹出,必须保留该节点的指针。
(3)此时,该判断此节点的右子树:
a.右子树为NULL,返回到栈顶;
b.右子树不为NULL,把该节点当根节点,重复(1)(2)(3)......
代码如下:void _InOrder(Node * root)
{
if (root == NULL)
{
return;
}
Node * cur = root;
stack s;
while (cur || !s.empty())
{
while (cur)//当没有左子树时,停止入栈
{
s.push(cur);
cur = cur->_left;
}
Node * top = s.top();//保留栈顶指针,判断是否有右子树
cout <_data <";
s.pop();
if (top->_right == NULL)//没有右子树时,不需要压栈
{
cur = NULL;
}
else//当存在右子树时,把右子树的根节点压入栈中,循环去判断该节点的左子树是否存在
{
cur = top->_right;
}
}
}
3.后序遍历
(1)一直入栈,一直到二叉树的最左边最下边的节点。
(2)按照后序遍历的特点:左子树->右子树->根节点,输出栈顶的元素,并且弹出,必须保留该节点的指针。
(3)此时,该判断此节点的右子树,如果存在右子树,把该节点当作根节点,重复(1)(2)(3)
代码如下:void _PostOrder(Node * root)
{
if (root == NULL)
{
return;
}
Node * cur = root;
Node * prev = NULL;
stack s;
while (cur || !s.empty())
{
while (cur)//当没有左子树时,停止入栈
{
s.push(cur);
cur = cur->_left;
}
Node * top = s.top();//保留栈顶指针,判断它的右子树是否为空或者已经出栈
if (top->_right == NULL || top->_right == prev)
{
cout <_data <";
s.pop();
prev = top;//保留出栈元素的指针
cur = NULL;
}
else//当存在右子树时,把右子树的根节点压入栈中,循环去判断该节点的左子树是否存在
{
cur = top->_right;
}
}
}