PAT 团体程序设计天梯赛-练习集 L2-006. 树的遍历【数据结构】

最新推荐文章于 2025-04-11 20:10:16 发布

The_Dawn_Star

最新推荐文章于 2025-04-11 20:10:16 发布

阅读量1.1k

点赞数

分类专栏： PAT 数据结构文章标签：二叉树搜索遍历数据结构

本文链接：https://blog.youkuaiyun.com/wlx65003/article/details/51456996

版权

数据结构同时被 2 个专栏收录

4 篇文章

订阅专栏

PAT

3 篇文章

订阅专栏

本文介绍了一种根据二叉树的后序和中序遍历结果，实现层次遍历输出的方法。通过寻找根节点，并在中序遍历中划分左右子树，最终实现了层次遍历的输出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目连接

https://www.patest.cn/contests/gplt/L2-006

思路

给你一棵二叉树的后序和中序遍历，叫你输出层次遍历。
首先找到根节点，就是后序的最后一个数。
然后分别在中序中找这个数的左边和右边，分别挑出在后序中排最后的数，注意遇到已经确定过的数要停止搜索。

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <unordered_map>
using namespace std;


int tree[50];
int post[50], in[50], vis[50];
unordered_map<int, int>order;
int main()
{

    int n;
    scanf("%d",&n);
    for(int i=1 ; i<=n ; ++i)
    {
        scanf("%d",&post[i]);
        order[post[i]]=i;
    }
    for(int i=1 ; i<=n ; ++i)
    {
        scanf("%d",&in[i]);
    }
    queue<int>q;
    q.push(post[n]);
    int cnt=1;
    for(int step=1 ; step<=n ; ++step)
    {
        int u=q.front(); q.pop();
        vis[u]=1;
        tree[cnt++]=u;
        for(int i=1 ; i<=n ; ++i)if(in[i]==u)
        {
            int max_order=0, last_num=-1;
            for(int j=i-1 ; j>=1 && !vis[in[j]] ; --j)
            {
                if(order[in[j]]>max_order)
                {
                    max_order=order[in[j]];
                    last_num=in[j];
                }
            }
            if(last_num!=-1) q.push(last_num);
            max_order=0; last_num=-1;
            for(int j=i+1 ; j<=n && !vis[in[j]] ; ++j)
            {
                if(order[in[j]]>max_order)
                {
                    max_order=order[in[j]];
                    last_num=in[j];
                }
            }
            if(last_num!=-1) q.push(last_num);
            break;
        }
    }
    for(int i=1 ; i<cnt ; ++i)
    {
        if(i==1) printf("%d", tree[i]);
        else printf(" %d", tree[i]);
    }
    printf("\n");
    return 0;
}