PAT(A)-1127. ZigZagging on a Tree (30)(树的重建)

记录一个菜逼的成长。。

题目链接

题目大意：
给你一颗二叉树的中序和后序。
要你输出指定顺序的序列。

又是树的重建。。
输出的序列跟层序差不多，判断一下是否逆序存放就行了。

PS：这次春季，水的一匹。。这树的重建出了不知道多少遍。。

#include <bits/stdc++.h>
using namespace std;
#define cl(a,b) memset(a,b,sizeof(a))
#define pb push_back
typedef long long LL;
const int INF = 0x3f3f3f3f;
map<int,int>tree;
vector<int>ans;
const int maxn = 35;
int post[maxn],in[maxn];
void build(int t,int l1,int r1,int l2,int r2)
{
    if(r1 < l1)return ;
    tree[t] = post[r2];
    int p = 0;
    for( int i = l1; i <= r1; i++ ){if(in[i] != tree[t])p++;else break;}
    build(t<<1,l1,l1+p-1,l2,l2+p-1);
    build(t<<1|1,l1+p+1,r1,l2+p,r2-1);
}
void bfs(int t)
{
    queue<int>q;
    vector<int>v;
    q.push(t);
    int cnt = 1,tmp = 0,flag = 0;
    while(!q.empty()){
        int f = q.front();q.pop();
        v.pb(tree[f]);
        if(cnt)cnt--;
        if(tree[f<<1])q.push(f<<1),tmp++;
        if(tree[f<<1|1])q.push(f<<1|1),tmp++;
        if(!cnt){
            if(flag){
                for( int i = 0; i < v.size(); i++ ){
                    ans.pb(v[i]);
                }
                flag = 0;
            }
            else {
                for( int i = v.size()-1; i >= 0; i-- ){
                    ans.pb(v[i]);
                }
                flag = 1;
            }
            v.clear();
            cnt = tmp,tmp = 0;
        }
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n)){
        tree.clear();
        for( int i = 0; i < n; i++ ){
            scanf("%d",in+i);
        }
        for( int i = 0; i < n; i++ ){
            scanf("%d",post+i);
        }
        build(1,0,n-1,0,n-1);
        bfs(1);
        for( int i = 0; i < ans.size(); i++ ){
            printf("%d%c",ans[i],i!=ans.size()-1?' ':'\n');
        }
    }
    return 0;
}