L2-006. 树的遍历

最新推荐文章于 2024-04-09 22:51:29 发布

Guuuuuu老师儿

最新推荐文章于 2024-04-09 22:51:29 发布

阅读量125

点赞数

CC 4.0 BY-SA版权

分类专栏： # 天梯赛

本文链接：https://blog.youkuaiyun.com/codeswarrior/article/details/79757924

天梯赛专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一种算法，该算法接收一棵二叉树的后序遍历和中序遍历序列作为输入，并输出该树的层序遍历序列。通过构建二叉树并使用队列实现广度优先搜索来完成层序遍历。文章提供了完整的C++代码实现。

L2-006. 树的遍历

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数N（<=30），是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。

输出格式：

在一行中输出该树的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

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

输出样例：

4 1 6 3 5 7 2

code

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <queue>
 #include <vector>
 using namespace std;
 int n;
 int post[50],in[50];
 struct node{
     int l,r;
 }tre[50];
 int build(int lp,int rp,int lin,int rin){
    if(lp > rp)
        return 0;
    int p1,p2;
    int rt = post[rp];
    p1 = lin;
    while(in[p1] != rt) p1++;
    p2 = p1 - lin;
    tre[rt].l = build(lp,lp+p2-1,lin,p1-1);
    tre[rt].r = build(lp+p2,rp-1,p1+1,rin);
    return rt;
 }
 void bfs(int rt){
     queue<int>q;
     vector<int>v;
     q.push(rt);
     while(!q.empty()){
        int w = q.front();
        q.pop();
        if(w == 0) break;
        v.push_back(w);
        if(tre[w].l != 0)
            q.push(tre[w].l);
        if(tre[w].r != 0)
            q.push(tre[w].r);
     }
     for(int i = 0; i < v.size(); i++){
        printf("%d%c",v[i],i == v.size()-1 ? '\n' : ' ');
     }
 }
 int main(){
     scanf("%d",&n);
     for(int i = 0; i < n; i++){
        scanf("%d",&post[i]);
     }
     for(int i = 0; i < n; i++){
        scanf("%d",&in[i]);
     }
     int root = build(0,n-1,0,n-1);
     bfs(root);
     return 0;
 }