PAT 团体程序设计天梯赛-练习集 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
);
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;
}