二叉树的遍历序列还原需要两个序列来确定,其中中序序列是必不可少的,这里介绍的是利用中序和后序序列还原,中序和前序还原方法类似。
给出两个序列
中序:badcfeg
后序:bdfgeca
如果子树非空的情况下,中序遍历的特点是先遍历左子树,再输出父节点,再遍历右子树,后序遍历的特点是先遍历左子树和右子树,最后输出父节点,整个过程是递归进行的。
根据这样的特点,我们可以得知,对于后序序列,最后一个输出的肯定是整棵树的根节点。而对于中序序列,每一个父节点的左边输出的节点一定是左子树上的节点,右边输出的节点一定是右子树上的节点。有了这样的特点,我们就可以把还原分成左子树和右子树来递归进行。
对于样例,还原步骤如下
图例:左子树 父节点 右子树
step1:从后往前遍历后序序列,第一个元素为a,说明为根节点
b a dcfeg
step2:第二个元素为c,a的右子树指向c
d c feg
step3:第三个元素为e,c的右子树指向e
f e g
step4:第四个元素为g,e的右子树指向g,没有左右子树了,停止。
这是dfs树右子树的一支,对左子树的操作类似。
经过这样一系列操作,最后就能复原整棵二叉树。
a
b c
d e
f g
#include <stdio.h>
#include <string.h>
char t1[60],t2[60];
int p;
int pos[255];
char tree[255][2];
int cou;
void deal(int l,int r,char head,int p){
int m,tmp;
if(p==0||cou>1000) return ;
m=pos[head];
tmp=pos[t2[p-1]];
if(tmp<l) return ;
if(m<tmp) tree[head][1]=t2[p-1];
if(p-(r-m)-1>=0){
tmp=pos[t2[p-(r-m)-1]];
if(tmp<l) return ;
if(m>tmp) tree[head][0]=t2[p-(r-m)-1];
}
cou++;
deal(m+1,r,t2[p-1],p-1);
deal(l,m-1,t2[p-(r-m)-1],p-(r-m)-1);
}
int main()
{
cou=0;
scanf("%s",t1);
scanf("%s",t2);
int len,i,j,nn=1,tmp[2][100],t=1,num;
char head;
len=strlen(t1);
head=t2[len-1];
p=len-1;
memset(tree,0,sizeof(tree));
for(i=0;i<len;i++) pos[t1[i]]=i;
deal(0,len-1,head,len-1);
tmp[1][0]=head;
num=0;
while(nn){
for(i=0;i<nn;i++){
head=tmp[t&1][i];
printf("%c",head);
if(tree[head][0]) tmp[!(t&1)][num++]=tree[head][0];
if(tree[head][1]) tmp[!(t&1)][num++]=tree[head][1];
}
nn=num;
num=0;
t++;
}
printf("\n");
return 0;
}