7-8 根据后序和中序遍历输出先序遍历 (25 分)
本题要求根据给定的一棵二叉树的后序遍历和中序遍历结果,输出该树的先序遍历结果。
输入格式:
第一行给出正整数N(≤30),是树中结点的个数。随后两行,每行给出N个整数,分别对应后序遍历和中序遍历结果,数字间以空格分隔。题目保证输入正确对应一棵二叉树。
输出格式:
在一行中输出Preorder:
以及该树的先序遍历结果。数字间有1个空格,行末不得有多余空格。
输入样例:
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7
输出样例:
Preorder: 4 1 3 2 6 5 7
基本思路就是利用后序排列可以不断找出根节点,从而在中序排列中找到该节点的位置,将中序排列拆分为左右两个子树成为新的中序排列,根据拆分结果可以将后序排列相同拆分为两个新的后序排列......如此递归求解直到子树为空。在这个递归过程中就可以不断按顺序找出各个根节点完成树的重建!
#include<stdio.h>
#include<stdlib.h>
typedef struct node{
int data;
struct node *l,*r;
}*bt;
//根据输入的中序后序排列来重建树
bt createTree(int *in,int *post,int n){ //指针in和post指向中后序的起始位置,n是长度
if(n<=0)
return NULL; //递归出口,返回空指针
int *p=in; //p是从中序排列中找到根的位置
while(p){
if(*p==*(post+n-1))
break;
p++;
}
int len=p-in; //计算长度
bt t=(bt)malloc(sizeof(struct node)); //创建这个节点并赋值,递归创建其左右孩子
t->data=*p;
t->l=createTree(in,post,len);
t->r=createTree(p+1,post+len,n-len-1); //原来长度为n,减去左边子树的一段和根节点,长度变为n-len-1
return t;
}
void preorder(bt t){ //先序遍历创建的树得到先序排列
if(t){
printf(" %d",t->data);
preorder(t->l);
preorder(t->r);
}
}
int main(){
int n,in[50],post[50];
bt t;
scanf("%d",&n);
for(int i = 0;i < n;i++)
scanf("%d",&post[i]);
for(int i = 0;i < n;i++)
scanf("%d",&in[i]);
t=createTree(in,post,n);
printf("Preorder:");
preorder(t);
return 0;
}
同理按照此思路也可以完成已知先序中序求解后序排列,只需要对重建树的代码进行更改:
bt createTree(int *in,int *pre,int n){
if(n<=0)
return NULL;
int *p=in;
while(p){
if(*p==*pre)
break;
p++;
}
int len=p-in;
bt t=(bt)malloc(sizeof(struct node));
t->data=*p;
t->l=createTree(in,pre+1,len); //注意选取位置的变化
t->r=createTree(p+1,pre+1+len,n-len-1);
}