本篇文章主要详解利用栈的方式二叉树先序、中序、后序遍历的非递归写法
首先我们需要实现一颗二叉树。
以下是通过先序序列建树的代码
例如:先序序列{1,2,4,10,'#','#',6,11,'#','#',7'#','#','#',3,8,'#','#',9}代表以下的树
typedef struct Node
{
int _data;
Node* _right;
Node* _left;
Node()
{}
Node(int x)
{
_data = x;
_left = NULL;
_right = NULL;
}
}*pNode;
//参数类型arr 数组指针,index 下标,size 数组元素个数,invalid代表先序遍历时遇到的NULL
pNode createTree_PrevOrder( const int* arr, size_t& index,const size_t size,int invalid){Node* root = NULL;if (index < size && arr[index] != invalid){root = new Node(arr[index]);root->_left = createTree_PrevOrder(arr, ++index, size, invalid);root->_right = createTree_PrevOrder(arr, ++index, size, invalid);}return root;}
先序非递归:
先序遍历时,每当我们压入一个结点,我们压入结点前对其进行访问。
void prevOrder_NR(pNode root)
{
if (NULL == root)
cout << "empty!" << endl;
stack<pNode> s;
pNode cur = root;
while (cur || !s.empty())
{
while (cur) //判断:cur为空,遍历栈顶结点的右子树结点,cur不为空;访问cur,并遍历cur的左子树结点
{
cout << cur->_data << "->";//遍历一个结点则进行访问
s.push(cur);
cur = cur->_left;
}
cur = s.top(); //此时cur为空,因此需要对cur进行处理
s.pop(); //将栈顶结点的右子树结点压入栈前,将栈顶结点弹出,避免死循环
cur = cur->_right;//在下一个while时将右子树结点压入栈
}
cout << "over"<<endl;
}
中序非递归:
中序时我们需要在遍历完左子树后访问根节点,再去遍历右子树,因此我们仅需依照先序遍历修改部分代码。
void inOrder_NR(pNode root)
{
if (NULL == root)
cout << "empty" << endl;
stack<pNode> s;
pNode cur = root;
while (cur || !s.empty())
{
while (cur)//压入左子树结点
{
s.push(cur);
cur = cur->_left;
}
cur = s.top();//取栈顶结点
cout << cur->_data << "->" ;//访问左结点,此时栈顶节点有两种情况:1.左子树结点为空;2.左子树结点已被访问弹出
s.pop(); //弹出访问过的结点
cur = cur->_right; //将栈顶结点的右子树结点入栈
}
cout << "over" << endl;
}
后序遍历:
后序遍历时由于访问完左右子树后才能访问根结点,因此需要将根结点在栈内保留到左右子树被访问后,但同时会出现一个问题,当右子树弹出后遇到根结点又会将右子树结点压入栈中,造成死循环,因此我们需要在定义一个变量last代表最后一个访问的结点,当last与栈顶结点的右子树结点相同时,则不再将右子树结点压入栈中。
void pastOrder_NR(pNode root)
{
if (NULL == root)
cout << "empty" << endl;
pNode cur = root;
pNode last = NULL;
stack<pNode> s;
while (cur || !s.empty())
{
while (cur)//压入左子树结点
{
s.push(cur);
cur = cur->_left;
}
cur = s.top();
if (cur->_right && last != cur->_right)//考虑栈顶结点的右子树结点。存在且没被访问过,将右子树结点压入栈中
{
cur = cur->_right;
}
else if ((NULL == cur->_right) || (last == cur->_right))
//右子树结点为空或者已经被访问过,则访问栈顶结点并弹出
{
cout << cur->_data << "->";
last = cur;//更新last值
s.pop();
//cur置空作用在于当原栈顶结点被访问并弹出后,下一层while是将当前栈顶结点的左子树入栈,当前栈顶结点的左子树已经被遍历过,
//因此会造成死循环,所以将cur置空,直接考虑当前栈顶结点的右子树
//一旦某个结点入栈,首先会遍历这个结点的左子树,然后考虑右子树的情况
cur = NULL;
}
}cout << "over" << endl;
}
附带递归写法用于测试正确性
void prevOrder(pNode root)
{
if (root == NULL)
return;
cout << root->_data << "->";
prevOrder(root->_left);
prevOrder(root->_right);
}
void pastOrder(pNode root)
{
if (root == NULL) return;
pastOrder(root->_left);
pastOrder(root->_right);
cout << root->_data << "->";
}
void inOrder(pNode root)
{
if (root == NULL) return;
inOrder(root->_left);
cout << root->_data << "->";
inOrder(root->_right);
}
测试结果:void test() { int arr[] = { 1, 2, 4, 10, '#', '#', 6, 11, '#', '#', 7,'#', '#', '#', 3, 8, '#', '#', 9 }; size_t index = 0; pNode root=createTree_PrevOrder(arr, index, sizeof(arr)/sizeof(arr[0]), '#'); cout << "prevOrder_NR:" ; prevOrder_NR(root); cout << "prevOrder :"; prevOrder(root); cout << endl; cout << "inOrder_NR:"; inOrder_NR(root); cout << "inOrder :"; inOrder(root); cout << endl; cout << "pastOrder_NR:"; pastOrder_NR(root); cout << "pastOrder :"; pastOrder(root); cout << endl; }