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

最新推荐文章于 2022-10-18 23:57:07 发布

oranges_c

最新推荐文章于 2022-10-18 23:57:07 发布

阅读量874

点赞数

CC 4.0 BY-SA版权

分类专栏： ------------acm------------ 树 PAT cccc-gplt 文章标签： acm cccc-gplt pat 数据结构树

本文链接：https://blog.youkuaiyun.com/oranges_c/article/details/60779136

------------acm------------ 同时被 3 个专栏收录

117 篇文章

订阅专栏

数据结构

34 篇文章

订阅专栏

PAT

21 篇文章

订阅专栏

本文介绍了一种通过给定的二叉树中序和后序遍历序列来重建该二叉树，并输出一种类似层序遍历的序列的方法。通过递归地构建二叉树并使用队列进行广度优先搜索，可以实现这种序列的输出。

记录一个菜逼的成长。。

题目链接

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

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

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