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