已知二叉树的先序序列、中序序列,试编写算法建立该二叉树,输出后序遍历以及层次遍历序列。
解析:用CreateBiTree(int pre_head, int pre_tail, int in_head, int in_tail)函数建立二叉树。其中pre_head,pre_tail为结点子树的先序序列范围,in_head,in_tail为结点子树的中序序列范围。在先序中确定根,然后在中序序列中确定根的左右子树序列,递归此过程。
例如:
先序:ABCDEFGHI
中序:BCAEDGHFI
找到树根A,则对于A:
左子树:T->lchild=CreatBiTree(1,2,0,1)//BC和BC
右子树:T->rchild=CreatBiTree(3,8,3,8)//DEFGHI和EDGHFI
再在A的左子树中对B递归,右子树中对D递归…
最后建立完成一颗完整的二叉树。
#include <bits/stdc++.h>
using namespace std;
//ABCDEFGHI
//BCAEDGHFI
#define MAXSIZE 10
typedef struct BiTNode
{
char data;
struct BiTNode *lchild,*rchild;
} Node,*BiTree;
typedef struct Queue
{
BiTree base[MAXSIZE];
int front,rear;
}Queue;
char pre[20],in[20];//先序,中序元素数组
BiTree CreateBiTree(int pre_head, int pre_tail, int in_head, int in_tail)
{
if(pre_head>pre_tail)//即len等于0
return NULL;
Node* node = new Node();
node->data = pre[pre_head];
int i;
for (i = in_head; i < in_tail; ++i)
{
if(in[i]==pre[pre_head])//中序序列中找到头
break;
}
int len = i-in_head;
node->lchild=CreateBiTree(pre_head + 1, pre_head + len, in_head, i - 1);
node->rchild=CreateBiTree(pre_head + len + 1, pre_tail, i + 1, in_tail);
return node;
}
void PostOrderTraverse(BiTree T)//后序遍历
{
if(T)
{
PostOrderTraverse(T->lchild);
PostOrderTraverse(T->rchild);
cout<<T->data;
}
else
return;
}
void LevelOrderTraverse(BiTree T)//层次遍历
{
BiTree p;
Queue Q;
Q.front=Q.rear=0;
if(T)
{
Q.base[Q.rear]=T;
Q.rear=(Q.rear+1)%MAXSIZE;
while(Q.front!=Q.rear)
{
p=Q.base[Q.front];
cout<<p->data;
Q.front=(Q.front+1)%MAXSIZE;
if(p->lchild)
{
Q.base[Q.rear]=p->lchild;
Q.rear=(Q.rear+1)%MAXSIZE;
}
if(p->rchild)
{
Q.base[Q.rear]=p->rchild;
Q.rear=(Q.rear+1)%MAXSIZE;
}//p的左右子树入队
}
}
}
int main()
{
BiTree T;
cout <<"输入先序序列:";
cin>>pre;
cout <<"输入中序序列:";
cin>>in;
int n=strlen(pre);
T=CreateBiTree(0,n-1,0,n-1);//根据先序和中序创建二叉树,初始左右范围0-(n-1)
cout<<"后序遍历序列为:";
PostOrderTraverse(T);
cout<<"\n层次遍历序列为:";
LevelOrderTraverse(T);
return 0;
}
//小绵杨Yeanling