在CU的blog中看到這篇還不錯(cuò)的文章,故轉(zhuǎn)來(lái)拜讀。
作者的后序遍歷寫(xiě)的不錯(cuò),用棧加上遍歷指針,這樣就不需要修改結(jié)點(diǎn)的數(shù)據(jù)結(jié)構(gòu)了。贊一個(gè)~
這里是源地址。
http://blog./space.php?uid=20775243&do=blog&id=2554952
/*
* Description:
* 二叉搜索樹(shù)的相關(guān)操作(創(chuàng)建,插入節(jié)點(diǎn),前、中、后序遞歸和非遞歸遍歷二叉樹(shù))
* Author :FinL
* Language: C
* Date : 2010-08-29
*/
#include<stdio.h>
#include<stdlib.h>
typedef struct Node{
int data;
Node *leftchild;
Node *rightchild;
}Node;
/*
初始化一棵二叉樹(shù)排序樹(shù)。
*/
void InitBinaryTree(Node**root,int elem)
{
*root=(Node*)malloc(sizeof(Node));
if(!(*root))
{
printf("Memory allocation for root failed.\n");
return;
}
(*root)->data=elem;
(*root)->leftchild=NULL;
(*root)->rightchild=NULL;
}
/*
向二叉樹(shù)排序樹(shù)中插入結(jié)點(diǎn)。
*/
void InsertNode(Node *root,int elem)
{
Node *newnode=NULL;
Node *p=root,*last_p=NULL;
newnode=(Node*)malloc(sizeof(Node));
if(!newnode)
{
printf("Memory allocation for newnode failed.\n");
return;
}
newnode->data=elem;
newnode->leftchild=NULL;
newnode->rightchild=NULL;
while(NULL!=p)
{
last_p=p;
if(newnode->data<p->data)
{
p=p->leftchild;
}
else if(newnode->data>p->data)
{
p=p->rightchild;
}
else
{
printf("Node to be inserted has existed.\n");
free(newnode);
return;
}
}
p=last_p;
if(newnode->data<p->data)
{
p->leftchild=newnode;
}
else
{
p->rightchild=newnode;
}
}
/*
創(chuàng)建一棵二叉樹(shù)排序樹(shù)。
*/
void CreatBinarySearchTree(Node **root,int data[],int num)
{
int i;
for(i=0;i<num;i++)
{
if(NULL==*root)
{
InitBinaryTree(root,data[i]);
}
else
{
InsertNode(*root,data[i]);
}
}
}
/*
前序遍歷二叉樹(shù),遞歸方法。
*/
void PreOrderRec(Node *root)
{
if(NULL!=root)
{
printf("%d ",root->data);
PreOrderRec(root->leftchild);
PreOrderRec(root->rightchild);
}
}
/*
前序遍歷二叉樹(shù),非遞歸方法。
*/
void PreOrderNoRec(Node *root)
{
Node *p=root;
Node *stack[30];
int num=0;
while(NULL!=p||num>0)
{
while(NULL!=p)
{
printf("%d ",p->data);
stack[num++]=p;
p=p->leftchild;
}
num--;
p=stack[num];
p=p->rightchild;
}
printf("\n");
}
/*
中序遍歷二叉樹(shù),遞歸方法。
*/
void InOrderRec(Node *root)
{
if(NULL!=root)
{
InOrderRec(root->leftchild);
printf("%d ",root->data);
InOrderRec(root->rightchild);
}
}
/*
中序遍歷二叉樹(shù),非遞歸方法,使用棧。
*/
void InOrderNoRec(Node *root)
{
Node *p=root;
int num=0;
Node *stack[30];
while(NULL!=p||num>0)
{
while(NULL!=p)
{
stack[num++]=p;
p=p->leftchild;
}
num--;
p=stack[num];
printf("%d ",p->data);
p=p->rightchild;
}
printf("\n");
}
/*
后序遍歷二叉樹(shù),遞歸方法。
*/
void PostOrderRec(Node *root)
{
if(NULL!=root)
{
PostOrderRec(root->leftchild);
PostOrderRec(root->rightchild);
printf("%d ",root->data);
}
}
/*
后序遍歷二叉樹(shù),非遞歸方法。
*/
void PostOrderNoRec(Node *root)
{
Node *p=root;
Node *stack[30];
int num=0;
Node *have_visited=NULL;
while(NULL!=p||num>0)
{
while(NULL!=p)
{
stack[num++]=p;
p=p->leftchild;
}
p=stack[num-1];
if(NULL==p->rightchild||have_visited==p->rightchild)
{
printf("%d ",p->data);
num--;
have_visited=p;
p=NULL;
}
else
{
p=p->rightchild;
}
}
printf("\n");
}
int main()
{
Node *root=NULL;
int num=0;
int data[]={5,2,4,0,0,5,0,0,3,6,8,0,10,0,7,0,9};
num=sizeof(data)/sizeof(int);
CreatBinarySearchTree(&root,data,num);
printf("This is Preorder traversal.\n");
PreOrderNoRec(root);
PreOrderRec(root);
printf("\n");
printf("This is Inorder traversal.\n");
InOrderNoRec(root);
InOrderRec(root);
printf("\n");
printf("This is Postorder traversal.\n");
PostOrderNoRec(root);
PostOrderRec(root);
printf("\n");
return 0;
}
另外一種方法是在stack結(jié)構(gòu)體里面添加一個(gè)標(biāo)志,用于判斷節(jié)點(diǎn)是否已經(jīng)遍歷過(guò)(或者說(shuō)是否需要從stack里面彈出)
前序、中序、后序的非遞歸遍歷中,要數(shù)后序最為麻煩,如果只在棧中保留指向結(jié)點(diǎn)的指針,那是不夠的,必須有一些額外的信息存放在棧中。
方法有很多,這里只舉一種,先定義棧結(jié)點(diǎn)的數(shù)據(jù)結(jié)構(gòu)
typedef struct{Node * p; int rvisited;}SNode //Node 是二叉樹(shù)的結(jié)點(diǎn)結(jié)構(gòu),rvisited==1代表p所指向的結(jié)點(diǎn)的右結(jié)點(diǎn)已被訪問(wèn)過(guò)。
lastOrderTraverse(BiTree bt){
//首先,從根節(jié)點(diǎn)開(kāi)始,往左下方走,一直走到頭,將路徑上的每一個(gè)結(jié)點(diǎn)入棧。
p = bt;
while(bt){
push(bt, 0); //push到棧中兩個(gè)信息,一是結(jié)點(diǎn)指針,一是其右結(jié)點(diǎn)是否被訪問(wèn)過(guò)
bt = bt.lchild;
}
//然后進(jìn)入循環(huán)體
while(!Stack.empty()){ //只要棧非空
sn = Stack.getTop(); // sn是棧頂結(jié)點(diǎn)
//注意,任意一個(gè)結(jié)點(diǎn)N,只要他有左孩子,則在N入棧之后,N的左孩子必然也跟著入棧了(這個(gè)體現(xiàn)在算法的后半部分),所以當(dāng)我們拿到棧頂元素的時(shí)候,可以確信這個(gè)元素要么沒(méi)有左孩子,要么其左孩子已經(jīng)被訪問(wèn)過(guò),所以此時(shí)我們就不關(guān)心它的左孩子了,我們只關(guān)心其右孩子。
//若其右孩子已經(jīng)被訪問(wèn)過(guò),或是該元素沒(méi)有右孩子,則由后序遍歷的定義,此時(shí)可以visit這個(gè)結(jié)點(diǎn)了。
if(!sn.p.rchild || sn.rvisited){
p = pop();
visit(p);
}
else //若它的右孩子存在且rvisited為0,說(shuō)明以前還沒(méi)有動(dòng)過(guò)它的右孩子,于是就去處理一下其右孩子。
{
//此時(shí)我們要從其右孩子結(jié)點(diǎn)開(kāi)始一直往左下方走,直至走到盡頭,將這條路徑上的所有結(jié)點(diǎn)都入棧。
//當(dāng)然,入棧之前要先將該結(jié)點(diǎn)的rvisited設(shè)成1,因?yàn)槠溆液⒆拥娜霔R馕吨挠液⒆颖貙⑾扔谒辉L問(wèn)(這很好理解,因?yàn)槲覀兛偸菑臈m斎〕鲈貋?lái)進(jìn)行visit)。由此可知,下一次該元素再處于棧頂時(shí),其右孩子必然已被visit過(guò)了,所以此處可以將rvisited設(shè)置為1。
sn.rvisited = 1;
//往左下方走到盡頭,將路徑上所有元素入棧
p = sn.p.rchild;
while(p != 0){
push(p, 0);
p = p.lchild;
}
}//這一輪循環(huán)已結(jié)束,剛剛?cè)霔5哪切┙Y(jié)點(diǎn)我們不必管它了,下一輪循環(huán)會(huì)將這些結(jié)點(diǎn)照顧的很好。
}
}