pat 团体赛练习题集 L2-006. 树的遍历

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

输入格式:

输入第一行给出一个正整数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

题目很简单,就是已知后序中序求层序。其实和已知后序中序求前序一样的做法,只要将求得的序列用数组保存起来,用数组存起来的序列就是层序遍历的结果。

//Asimple
#include <bits/stdc++.h>
#define INF 1e10
#define mod 10007
#define swap(a,b,t) t = a, a = b, b = t
#define CLS(a, v) memset(a, v, sizeof(a))
#define debug(a) cout << #a << " = " << a <<endl
#define abs(x) x<0?-x:x
using namespace std;
typedef long long ll;
const int maxn = ;
int n;
int pos[maxn], in[maxn];
vector<int> res(, -); void solve(int root, int s, int e, int index) {
if( s > e ) return ;
int i = s;
while( i<e && in[i]!=pos[root])i++;
res[index] = pos[root];
solve(root--e+i, s, i-, index*+);
solve(root-, i+, e, index*+);
} void input() {
cin >> n;
for(int i=; i<n; i++) cin >> pos[i] ;
for(int i=; i<n; i++) cin >> in[i];
solve(n-, , n-, );
int cnt = ;
for(int i=; i<res.size(); i++) {
if( res[i]!=- && cnt!=n- ) {
printf("%d ", res[i]);
cnt ++;
} else if(res[i]!=-){
printf("%d\n", res[i]);
break;
}
}
} int main() {
input();
return ;
}
上一篇:浅析JNI函数的注册过程


下一篇:PAT L2-008 最长对称子串(模拟字符串)