浙大模拟（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; }