浙大模拟（1）问题 C: 还原二叉树

题目描述

给一棵二叉树的层序遍历序列和中序遍历序列，求这棵二叉树的先序遍历序列和后序遍历序列。

输入

每个输入文件中一组数据。

第一行一个正整数N（1<=N<=30），代表二叉树的结点个数（结点编号为1~N）。接下来两行，每行N个正整数，分别代表二叉树的层序遍历序列和中序遍历序列。数据保证序列中1~N的每个数出现且只出现一次。

输出

输出两行，每行N个正整数，分别代表二叉树的先序遍历序列和后序遍历序列。每行末尾不输出额外的空格。

样例输入

7
3 5 4 2 6 7 1
2 5 3 6 4 7 1

样例输出

3 5 2 4 6 7 1
2 5 6 1 7 4 3

#include<stdio.h>
#include<stdlib.h>
using namespace std;

int post[32];
int level[32];
int pre[32];
int in[32];
int n;
typedef struct BiTNode{
int key;
struct BiTNode *lchild;
struct BiTNode *rchild;
}BiTNode,*BiTree;

BiTree creatTree(BiTree &T,int l,int r){
if(l>r) return NULL;
int mid;
bool flag = false;
for(int i=1;i<=n;i++ ){
if(flag == false){
for(int j=l;j<=r;j++){
if(level[i]==in[j]){
mid = j;
flag=true;
break;
}
}
}
}

T = (BiTree)malloc(sizeof(BiTNode));
T->key = in[mid];
//	printf("mid=%d\n",mid);
T->lchild = creatTree(T->lchild,l,mid-1);
T->rchild = creatTree(T->rchild,mid+1,r);
return T;
}
int flag;
void preOrder(BiTree T){
if(T){
if(flag == false){
printf("%d",T->key);
flag = true;
}
else{
printf(" %d",T->key);
}
preOrder(T->lchild);
preOrder(T->rchild);
}
}

void postOrder(BiTree T){
if(T){
postOrder(T->lchild);
postOrder(T->rchild);
if(flag == false){
printf("%d",T->key);
flag = true;
}
else{
printf(" %d",T->key);
}
}
}

int main(){
while(scanf("%d",&n)!=EOF){
for(int i=1;i<=n;i++){
scanf("%d",&level[i]);
}
for(int i=1;i<=n;i++){
scanf("%d",&in[i]);
}
BiTree T ;
T = creatTree(T,1,n);
flag = false;
preOrder(T);
printf("\n");
flag = false;
postOrder(T);
}
return 0;
}