二叉樹遍歷非遞歸 C++實現(xiàn)代碼

來源：本站原創(chuàng)|時間：2020-01-10|欄目：C語言|點擊：次

二叉樹的非遞歸遍歷
二叉樹是一種非常重要的數(shù)據(jù)結(jié)構(gòu)，很多其它數(shù)據(jù)結(jié)構(gòu)都是基于二叉樹的基礎(chǔ)演變而來的。對于二叉樹，有前序、中序以及后序三種遍歷方法。因為樹的定義本身就是遞歸定義，因此采用遞歸的方法去實現(xiàn)樹的三種遍歷不僅容易理解而且代碼很簡潔。而對于樹的遍歷若采用非遞歸的方法，就要采用棧去模擬實現(xiàn)。在三種遍歷中，前序和中序遍歷的非遞歸算法都很容易實現(xiàn)，非遞歸后序遍歷實現(xiàn)起來相對來說要難一點。

一.前序遍歷

前序遍歷按照“根結(jié)點-左孩子-右孩子”的順序進行訪問。

1.遞歸實現(xiàn)

復(fù)制代碼代碼如下:

void preOrder1(BinTree *root) //遞歸前序遍歷 
{
 if(root!=NULL)
{
cout<<root->data<<" ";
preOrder1(root->lchild);
preOrder1(root->rchild);
}
}  

2.非遞歸實現(xiàn)
根據(jù)前序遍歷訪問的順序，優(yōu)先訪問根結(jié)點，然后再分別訪問左孩子和右孩子。即對于任一結(jié)點，其可看做是根結(jié)點，因此可以直接訪問，訪問完之后，若其左孩子不為空，按相同規(guī)則訪問它的左子樹；當(dāng)訪問其左子樹時，再訪問它的右子樹。因此其處理過程如下：

對于任一結(jié)點P：

1)訪問結(jié)點P，并將結(jié)點P入棧;

2)判斷結(jié)點P的左孩子是否為空，若為空，則取棧頂結(jié)點并進行出棧操作，并將棧頂結(jié)點的右孩子置為當(dāng)前的結(jié)點P，循環(huán)至1);若不為空，則將P的左孩子置為當(dāng)前的結(jié)點P;

3)直到P為NULL并且棧為空，則遍歷結(jié)束。

復(fù)制代碼代碼如下:

void preOrder2(BinTree *root) //非遞歸前序遍歷 
{
stack<BinTree*> s;
BinTree *p=root;
 while(p!=NULL||!s.empty())
{
 while(p!=NULL)
{
cout<<p->data<<" ";
s.push(p);
p=p->lchild;
}
 if(!s.empty())
{
p=s.top();
s.pop();
p=p->rchild;
}
}
}

二.中序遍歷

中序遍歷按照“左孩子-根結(jié)點-右孩子”的順序進行訪問。

1.遞歸實現(xiàn)

復(fù)制代碼代碼如下:

void inOrder1(BinTree *root) //遞歸中序遍歷
{
 if(root!=NULL)
{
inOrder1(root->lchild);
cout<<root->data<<" ";
inOrder1(root->rchild);
}
} 

2.非遞歸實現(xiàn)

根據(jù)中序遍歷的順序，對于任一結(jié)點，優(yōu)先訪問其左孩子，而左孩子結(jié)點又可以看做一根結(jié)點，然后繼續(xù)訪問其左孩子結(jié)點，直到遇到左孩子結(jié)點為空的結(jié)點才進行訪問，然后按相同的規(guī)則訪問其右子樹。因此其處理過程如下：

對于任一結(jié)點P，

1)若其左孩子不為空，則將P入棧并將P的左孩子置為當(dāng)前的P，然后對當(dāng)前結(jié)點P再進行相同的處理；

2)若其左孩子為空，則取棧頂元素并進行出棧操作，訪問該棧頂結(jié)點，然后將當(dāng)前的P置為棧頂結(jié)點的右孩子；

3)直到P為NULL并且棧為空則遍歷結(jié)束

復(fù)制代碼代碼如下:

void inOrder2(BinTree *root) //非遞歸中序遍歷
{
stack<BinTree*> s;
BinTree *p=root;
 while(p!=NULL||!s.empty())
{
 while(p!=NULL)
{
s.push(p);
p=p->lchild;
}
 if(!s.empty())
{
p=s.top();
cout<<p->data<<" ";
s.pop();
p=p->rchild;
}
} 
} 

三.后序遍歷

后序遍歷按照“左孩子-右孩子-根結(jié)點”的順序進行訪問。

1.遞歸實現(xiàn)

復(fù)制代碼代碼如下:

void postOrder1(BinTree *root) //遞歸后序遍歷
{
 if(root!=NULL)
{
postOrder1(root->lchild);
postOrder1(root->rchild);
cout<<root->data<<" ";
} 
} 

2.非遞歸實現(xiàn)

后序遍歷的非遞歸實現(xiàn)是三種遍歷方式中最難的一種。因為在后序遍歷中，要保證左孩子和右孩子都已被訪問并且左孩子在右孩子前訪問才能訪問根結(jié)點，這就為流程的控制帶來了難題。下面介紹兩種思路。

第一種思路：對于任一結(jié)點P，將其入棧，然后沿其左子樹一直往下搜索，直到搜索到?jīng)]有左孩子的結(jié)點，此時該結(jié)點出現(xiàn)在棧頂，但是此時不能將其出棧并訪問，因此其右孩子還為被訪問。所以接下來按照相同的規(guī)則對其右子樹進行相同的處理，當(dāng)訪問完其右孩子時，該結(jié)點又出現(xiàn)在棧頂，此時可以將其出棧并訪問。這樣就保證了正確的訪問順序。可以看出，在這個過程中，每個結(jié)點都兩次出現(xiàn)在棧頂，只有在第二次出現(xiàn)在棧頂時，才能訪問它。因此需要多設(shè)置一個變量標(biāo)識該結(jié)點是否是第一次出現(xiàn)在棧頂。

復(fù)制代碼代碼如下:

void postOrder2(BinTree *root) //非遞歸后序遍歷
{
stack<BTNode*> s;
BinTree *p=root;
BTNode *temp;
 while(p!=NULL||!s.empty())
{
 while(p!=NULL) //沿左子樹一直往下搜索，直至出現(xiàn)沒有左子樹的結(jié)點 
{
BTNode *btn=(BTNode *)malloc(sizeof(BTNode));
btn->btnode=p;
btn->isFirst=true;
s.push(btn);
p=p->lchild;
}
 if(!s.empty())
{
temp=s.top();
s.pop();
 if(temp->isFirst==true) //表示是第一次出現(xiàn)在棧頂 
{
temp->isFirst=false;
s.push(temp);
p=temp->btnode->rchild; 
}
 else //第二次出現(xiàn)在棧頂 
{
cout<<temp->btnode->data<<" ";
p=NULL;
}
}
} 
} 

第二種思路：要保證根結(jié)點在左孩子和右孩子訪問之后才能訪問，因此對于任一結(jié)點P，先將其入棧。如果P不存在左孩子和右孩子，則可以直接訪問它；或者P 存在左孩子或者右孩子，但是其左孩子和右孩子都已被訪問過了，則同樣可以直接訪問該結(jié)點。若非上述兩種情況，則將P的右孩子和左孩子依次入棧，這樣就保證了每次取棧頂元素的時候，左孩子在右孩子前面被訪問，左孩子和右孩子都在根結(jié)點前面被訪問。

復(fù)制代碼代碼如下:

void postOrder3(BinTree *root) //非遞歸后序遍歷
{
stack<BinTree*> s;
BinTree *cur; //當(dāng)前結(jié)點 
BinTree *pre=NULL; //前一次訪問的結(jié)點 
s.push(root);
 while(!s.empty())
{
cur=s.top();
 if((cur->lchild==NULL&&cur->rchild==NULL)||
(pre!=NULL&&(pre==cur->lchild||pre==cur->rchild)))
{
cout<<cur->data<<" "; //如果當(dāng)前結(jié)點沒有孩子結(jié)點或者孩子節(jié)點都已被訪問過 
s.pop();
pre=cur; 
}
 else
{
 if(cur->rchild!=NULL)
s.push(cur->rchild);
 if(cur->lchild!=NULL) 
s.push(cur->lchild);
}
} 
} 

四.整個程序完整的代碼

復(fù)制代碼代碼如下:

#include <iostream>
#include<string.h>
#include<stack>
using namespace std;

typedef struct node
{
char data;
struct node *lchild,*rchild;
}BinTree;

typedef struct node1
{
BinTree *btnode;
bool isFirst;
}BTNode;

void creatBinTree(char *s,BinTree *&root) //創(chuàng)建二叉樹，s為形如A(B,C(D,E))形式的字符串
{
int i;
bool isRight=false;
stack<BinTree*> s1; //存放結(jié)點
stack<char> s2; //存放分隔符
BinTree *p,*temp;
root->data=s[0];
root->lchild=NULL;
root->rchild=NULL;
s1.push(root);
i=1;
while(i<strlen(s))
{
if(s[i]=='(')
{
s2.push(s[i]);
isRight=false;
}
else if(s[i]==',')
{
isRight=true;
}
else if(s[i]==')')
{
s1.pop();
s2.pop();
}
else if(isalpha(s[i]))
{
p=(BinTree *)malloc(sizeof(BinTree));
p->data=s[i];
p->lchild=NULL;
p->rchild=NULL;
temp=s1.top();
if(isRight==true)
{
temp->rchild=p;
cout<<temp->data<<"的右孩子是"<<s[i]<<endl;
}
else
{
temp->lchild=p;
cout<<temp->data<<"的左孩子是"<<s[i]<<endl;
}
if(s[i+1]=='(')
s1.push(p);
}
i++;
}
}

void display(BinTree *root) //顯示樹形結(jié)構(gòu)
{
if(root!=NULL)
{
cout<<root->data;
if(root->lchild!=NULL)
{
cout<<'(';
display(root->lchild);
}
if(root->rchild!=NULL)
{
cout<<',';
display(root->rchild);
cout<<')';
}
}
}

void preOrder1(BinTree *root) //遞歸前序遍歷
{
if(root!=NULL)
{
cout<<root->data<<" ";
preOrder1(root->lchild);
preOrder1(root->rchild);
}
}

void inOrder1(BinTree *root) //遞歸中序遍歷
{
if(root!=NULL)
{
inOrder1(root->lchild);
cout<<root->data<<" ";
inOrder1(root->rchild);
}
}

void postOrder1(BinTree *root) //遞歸后序遍歷
{
if(root!=NULL)
{
postOrder1(root->lchild);
postOrder1(root->rchild);
cout<<root->data<<" ";
}
}

void preOrder2(BinTree *root) //非遞歸前序遍歷
{
stack<BinTree*> s;
BinTree *p=root;
while(p!=NULL||!s.empty())
{
while(p!=NULL)
{
cout<<p->data<<" ";
s.push(p);
p=p->lchild;
}
if(!s.empty())
{
p=s.top();
s.pop();
p=p->rchild;
}
}
}

void inOrder2(BinTree *root) //非遞歸中序遍歷
{
stack<BinTree*> s;
BinTree *p=root;
while(p!=NULL||!s.empty())
{
while(p!=NULL)
{
s.push(p);
p=p->lchild;
}
if(!s.empty())
{
p=s.top();
cout<<p->data<<" ";
s.pop();
p=p->rchild;
}
}
}

void postOrder2(BinTree *root) //非遞歸后序遍歷
{
stack<BTNode*> s;
BinTree *p=root;
BTNode *temp;
while(p!=NULL||!s.empty())
{
while(p!=NULL) //沿左子樹一直往下搜索，直至出現(xiàn)沒有左子樹的結(jié)點
{
BTNode *btn=(BTNode *)malloc(sizeof(BTNode));
btn->btnode=p;
btn->isFirst=true;
s.push(btn);
p=p->lchild;
}
if(!s.empty())
{
temp=s.top();
s.pop();
if(temp->isFirst==true) //表示是第一次出現(xiàn)在棧頂
{
temp->isFirst=false;
s.push(temp);
p=temp->btnode->rchild;
}
else //第二次出現(xiàn)在棧頂
{
cout<<temp->btnode->data<<" ";
p=NULL;
}
}
}
}

void postOrder3(BinTree *root) //非遞歸后序遍歷
{
stack<BinTree*> s;
BinTree *cur; //當(dāng)前結(jié)點
BinTree *pre=NULL; //前一次訪問的結(jié)點
s.push(root);
while(!s.empty())
{
cur=s.top();
if((cur->lchild==NULL&&cur->rchild==NULL)||
(pre!=NULL&&(pre==cur->lchild||pre==cur->rchild)))
{
cout<<cur->data<<" "; //如果當(dāng)前結(jié)點沒有孩子結(jié)點或者孩子節(jié)點都已被訪問過
s.pop();
pre=cur;
}
else
{
if(cur->rchild!=NULL)
s.push(cur->rchild);
if(cur->lchild!=NULL)
s.push(cur->lchild);
}
}
}

int main(int argc, char *argv[])
{
char s[100];
while(scanf("%s",s)==1)
{
BinTree *root=(BinTree *)malloc(sizeof(BinTree));
creatBinTree(s,root);
display(root);
cout<<endl;
preOrder2(root);
cout<<endl;
inOrder2(root);
cout<<endl;
postOrder2(root);
cout<<endl;
postOrder3(root);
cout<<endl;
}
return 0;
}