Problem Description
给一棵二叉树的层序遍历序列和中序遍历序列,求这棵二叉树的先序遍历序列和后序遍历序列。
Input
每个输入文件中一组数据。
第一行一个正整数N(1<=N<=30),代表二叉树的结点个数(结点编号为1~N)。接下来两行,每行N个正整数,分别代表二叉树的层序遍历序列和中序遍历序列。数据保证序列中1~N的每个数出现且只出现一次。
Output
输出两行,每行N个正整数,分别代表二叉树的先序遍历序列和后序遍历序列。每行末尾不输出额外的空格。
Sample Input
7
3 5 4 2 6 7 1
2 5 3 6 4 7 1
Sample Output
3 5 2 4 6 7 1
2 5 6 1 7 4 3
Author
Shoutmon
Source
15浙大考研机试模拟赛
#include <iostream>
#include <vector>
using namespace std;
int lev[32],ino[32],n;
vector<int> pre,pos;
void build(vector<int> lev,int ino[],int L,int R)
{
if(L>R) return;
int e=lev[0],idx=L,isLeft[32]={0};
while(e!=ino[idx]) isLeft[ino[idx++]]=1;
vector<int> left,right;
for(int i=1;i<(int)lev.size();++i)
{
if(isLeft[lev[i]]) left.push_back(lev[i]);
else right.push_back(lev[i]);
}
pre.push_back(e);
build(left,ino,L,idx-1);
build(right,ino,idx+1,R);
pos.push_back(e);
}
void display(vector<int> &v)
{
int n=v.size();
for(int i=0;i<n;++i)
i==n-1?printf("%d\n",v[i]):printf("%d ",v[i]);
}
int main()
{
scanf("%d",&n);
for(int i=0;i<n;++i)
scanf("%d",&lev[i]);
for(int i=0;i<n;++i)
scanf("%d",&ino[i]);
build(vector<int>(lev,lev+n),ino,0,n-1);
display(pre);
display(pos);
return 0;
}