【PAT甲级】1020 Tree Traversals (25 分)（树知二求一）

题意：

输入一个正整数N（N<=30），给出一棵二叉树的后序遍历和中序遍历，输出它的层次遍历。

trick：

当30个点构成一条单链时，如代码开头处的数据，大约1e9左右的结点编号大小，故采用结构体储存结点的序号（1~N），编号（代表它在完全二叉树上的位置）和值。PTA网站上可以用大小为31的数组存放结点，可能数据为一颗比较平衡的二叉树。

AAAAAccepted code：

/*
30
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
30 29 28 27 26 25 24 23 22 21 20 19 18 17 16 15 14 13 12 11 10 9 8 7 6 5 4 3 2 1
*/

 #define HAVE_STRUCT_TIMESPEC
 #include<bits/stdc++.h>
 using namespace std;
 int a[],b[];
 typedef struct node{
     int data;
     node *lchild,*rchild;
 }tree;
 tree *build(int l,int r,int L,int R){
     if(l>r)
         return NULL;
     tree *temp=new tree();
     temp->data=a[r];
     int i;
     for(i=L;i<=R;++i)
         if(b[i]==a[r])
             break;
     temp->lchild=build(l,l+i-L-,L,i-);
     temp->rchild=build(l+i-L,r-,i+,R);
     return temp;
 }
 int main(){
     ios::sync_with_stdio(false);
     cin.tie(NULL);
     cout.tie(NULL);
     int n;
     cin>>n;
     for(int i=;i<=n;++i)
         cin>>a[i];
     for(int i=;i<=n;++i)
         cin>>b[i];
     tree *temp=build(,n,,n);
     queue<tree *>q;
     q.push(temp);
     int flag=;
     while(!q.empty()){
         tree *now=q.front();
         q.pop();
         if(flag)
             cout<<" ";
         cout<<now->data;
         flag=;
         if(now->lchild)
             q.push(now->lchild);
         if(now->rchild)
             q.push(now->rchild);
     }
     return ;
 }

 #define HAVE_STRUCT_TIMESPEC
 #include<bits/stdc++.h>
 using namespace std;
 int pos[],in[];
 typedef struct lv{
     int index,num;
 };
 lv level[];
 int cnt=;
 void find_(int iroot,int istart,int iend,int index){
     if(istart>iend)
         return;
     level[++cnt].index=index;
     level[cnt].num=pos[iroot];
     int i=istart;
     while(in[i]!=pos[iroot])
         ++i;
     find_(iroot--iend+i,istart,i-,*index+);
     find_(iroot-,i+,iend,*index+);
 }
 bool cmp(lv a,lv b){
     if(a.index!=b.index)
         return a.index<b.index;
 }
 int main(){
     int n;
     cin>>n;
     for(int i=;i<n;++i)
         cin>>pos[i];
     for(int i=;i<n;++i)
         cin>>in[i];
     find_(n-,,n-,);
     sort(level+,level++n,cmp);
     for(int i=;i<cnt;++i)
         cout<<level[i].num<<" ";
     cout<<level[cnt].num;
     return ;
 }