中序遍历的非递归算法
话不多说直接上代码:
Status InOrderTraverse(BiTree T,Status(*Visit)(TElemType e)){
InitStack(S);Push(S,T);//根指针进栈
while(!StackEmpty(S)){
while(GetTop(S,p)&&p) Push(S,p->lchild);//代码标记1
Pop(S,p);//代码标记2
if(!StackEmpty(S)){
Pop(S,p);if(!Visit(p->data)) return ERROR;//代码标记4
Push(S,p->rchild);//代码标记5
}
}
return OK;
}
代码解释:
首先我们初始化一个栈,让根指针进栈。因为是中序遍历,所以我们首先要找到树的最左边结点,代码标记1完成的就是这个任务。那么代码标记1循环停止的条件不满足时,这个时候GetTop(S,p)得到的指针p是空的,因为到达最左边了,p->lchild是空,故我们需要把这个进栈的空指针给Pop掉,这是代码标记2的作用。
下面是关键一步了,我们需要访问当前栈顶结点了。Pop(S,p)删除栈顶结点并赋给p,然后Visit函数代表我们要对这个结点进行的 操作,这是代码标记4的作用。至于代码标记5就很明显了,将右结点进栈,重新进行这个操作,即:先找最左边结点。。。。
我们再用一个实例来捋一遍:
根据咱们的代码,第一遍循环:首先会找到B结点,执行visit,然后C进栈,此时栈里有A,C两个结点,第二遍循环:然后找到C结点,C出栈执行visit,然后右结点为空,进栈,第三遍循环:然后由于这个栈顶元素是空的,Pop掉,此时栈不空,还剩A,出栈,执行visit,D进栈,然后第四遍循环:D出栈,E进栈,E的左结点为空,进栈,然后出栈,E再出栈,执行visit,然后E的右空结点进栈,第五次循环:空结点出栈然后为空,pop掉之后,再出栈D,执行visit,然后F进栈,第六次循环:F的空结点进栈,被pop掉,F出栈,然后执行visit,F的空结点进栈,第七次循环:然后因为两次栈不为空,循环结束。
从这个过程我们发现,这个算法最本质的操作:
没有什么难的,都是必要的操作。