还原二叉树

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;
}