1020树的遍历

black_blank

于 2024-08-08 17:50:10 发布

阅读量210

点赞数 6

分类专栏： pta练习 acwing练习文章标签：算法数据结构

本文链接：https://blog.csdn.net/black_blank/article/details/141033049

版权

acwing练习同时被 2 个专栏收录

23 篇文章 0 订阅

订阅专栏

pta练习

4 篇文章 0 订阅

订阅专栏

1497. 树的遍历

一个二叉树，树中每个节点的权值互不相同。

现在给出它的后序遍历和中序遍历，请你输出它的层序遍历。

输入格式

第一行包含整数 NN，表示二叉树的节点数。

第二行包含 NN 个整数，表示二叉树的后序遍历。

第三行包含 NN 个整数，表示二叉树的中序遍历。

输出格式

输出一行 NN 个整数，表示二叉树的层序遍历。

数据范围

1≤N≤301≤N≤30,
官方并未给出各节点权值的取值范围，为方便起见，在本网站范围取为 1∼N1∼N。

输入样例：

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

输出样例：

4 1 6 3 5 7 2

#include <iostream>
#include <queue>
#include <map>
using namespace std;
const int N=40;
int inorder[N],postorder[N];
map<int,int>l,r,pos;
int build(int il,int ir,int pl,int pr)
{
    int root=postorder[pr];
    int k=pos[root];
    if(il<k)    l[root]=build(il,k-1,pl,pl+k-1-il);
    if(ir>k)    r[root]=build(k+1,ir,pl+k-il,pr-1);
    return root;
}
void bfs(int root)
{
    queue<int>q;
    q.push(root);
    while(!q.empty())
    {
        auto t=q.front();
        cout<<t;
        q.pop();
        if(l.count(t))  q.push(l[t]);
        if(r.count(t))  q.push(r[t]);
        if(!q.empty())   cout<<" ";    
    }
}
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)   cin>>postorder[i];
    for(int i=1;i<=n;i++)   cin>>inorder[i],pos[inorder[i]]=i;
    int root=build(1,n,1,n);
    bfs(root);

}